o
    ˜à·iu  ã                   @   s8   d dl Z d dlmZmZ d dlmZ G dd„ deƒZdS )é    N)Úadd_cli_argsÚmain)ÚBenchmarkSubcommandBasec                   @   sH   e Zd ZdZdZdZedejddfdd„ƒZ	e
d	ejddfd
d„ƒZdS )ÚBenchmarkSweepSubcommandz(The `sweep` subcommand for `vllm bench`.Úsweepz Benchmark for a parameter sweep.ÚparserÚreturnNc                 C   s   t |ƒ d S ©N)r   )Úclsr   © r   úZ/home/ubuntu/vllm_env/lib/python3.10/site-packages/vllm/entrypoints/cli/benchmark/sweep.pyr      ó   z%BenchmarkSweepSubcommand.add_cli_argsÚargsc                 C   s   t | ƒ d S r	   )r   )r   r   r   r   Úcmd   r   zBenchmarkSweepSubcommand.cmd)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚnameÚhelpÚclassmethodÚargparseÚArgumentParserr   ÚstaticmethodÚ	Namespacer   r   r   r   r   r   	   s    r   )r   Úvllm.benchmarks.sweep.clir   r   Ú#vllm.entrypoints.cli.benchmark.baser   r   r   r   r   r   Ú<module>   s   