[bugfix] fix the wrong parser (#17958)

Signed-off-by: reidliu41 <reid201711@gmail.com>
Co-authored-by: reidliu41 <reid201711@gmail.com>
This commit is contained in:
Reid 2025-05-12 12:58:02 +08:00 committed by GitHub
parent 08bf784078
commit ada50aa295
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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):
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]: