o
    i                     @   s   d dl Z d dlmZ ddlmZ ddlmZ ddlmZ ddlmZ	 ddl
mZ ddl
mZ ddlmZ ddlmZ dd	lmZ ddlmZ eefeefeefeefee	ffZd
e jfddZde jfddZdS )    N)VLLM_SUBCMD_PARSER_EPILOG   )SweepPlotArgs)main)SweepPlotParetoArgs)SweepServeArgs)SweepServeSLAArgs)SweepStartupArgsparserc                 C   sj   | j ddd}tD ])\}}|j|j|jd|j dd}|j|d || tjd|j d	|_	q	d S )
NT
sweep_type)requireddestzvllm bench sweep z
 [options])descriptionusagedispatch_functionzsweep )subcmd)
add_subparsersSUBCOMMANDS
add_parserparser_nameparser_helpset_defaultsadd_cli_argsr   formatepilog)r
   
subparserscmd
entrypointcmd_subparser r    O/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm/benchmarks/sweep/cli.pyr      s   


r   argsc                 C   s   |  |  d S )Nr   )r"   r    r    r!   r   +   s   r   )argparsevllm.entrypoints.utilsr   plotr   r   	plot_mainplot_paretor   plot_pareto_mainserver   
serve_main	serve_slar   serve_sla_mainstartupr	   startup_mainr   ArgumentParserr   	Namespacer    r    r    r!   <module>   s(   	