Unverified Commit ada50aa2 authored by Reid's avatar Reid Committed by GitHub
Browse files

[bugfix] fix the wrong parser (#17958)


Signed-off-by: default avatarreidliu41 <reid201711@gmail.com>
Co-authored-by: default avatarreidliu41 <reid201711@gmail.com>
parent 08bf7840
...@@ -4,12 +4,11 @@ import argparse ...@@ -4,12 +4,11 @@ import argparse
from vllm.collect_env import main as collect_env_main from vllm.collect_env import main as collect_env_main
from vllm.entrypoints.cli.types import CLISubcommand from vllm.entrypoints.cli.types import CLISubcommand
from vllm.entrypoints.openai.cli_args import make_arg_parser
from vllm.utils import FlexibleArgumentParser from vllm.utils import FlexibleArgumentParser
class CollectEnvSubcommand(CLISubcommand): class CollectEnvSubcommand(CLISubcommand):
"""The `serve` subcommand for the vLLM CLI. """ """The `collect-env` subcommand for the vLLM CLI. """
def __init__(self): def __init__(self):
self.name = "collect-env" self.name = "collect-env"
...@@ -23,12 +22,12 @@ class CollectEnvSubcommand(CLISubcommand): ...@@ -23,12 +22,12 @@ class CollectEnvSubcommand(CLISubcommand):
def subparser_init( def subparser_init(
self, self,
subparsers: argparse._SubParsersAction) -> FlexibleArgumentParser: subparsers: argparse._SubParsersAction) -> FlexibleArgumentParser:
serve_parser = subparsers.add_parser( collect_env_parser = subparsers.add_parser(
"collect-env", "collect-env",
help="Start collecting environment information.", help="Start collecting environment information.",
description="Start collecting environment information.", description="Start collecting environment information.",
usage="vllm collect-env") usage="vllm collect-env")
return make_arg_parser(serve_parser) return collect_env_parser
def cmd_init() -> list[CLISubcommand]: def cmd_init() -> list[CLISubcommand]:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment