o
    
Û¾iŒ  ã                   @   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 )ÚBenchmarkThroughputSubcommandz-The `throughput` subcommand for `vllm bench`.Ú
throughputz'Benchmark offline inference throughput.ÚparserÚreturnNc                 C   s   t |ƒ d S ©N)r   )Úclsr   © r   ú]/home/ubuntu/.local/lib/python3.10/site-packages/vllm/entrypoints/cli/benchmark/throughput.pyr      ó   z*BenchmarkThroughputSubcommand.add_cli_argsÚargsc                 C   s   t | ƒ d S r	   )r   )r   r   r   r   Úcmd   r   z!BenchmarkThroughputSubcommand.cmd)Ú__name__Ú
__module__Ú__qualname__Ú__doc__ÚnameÚhelpÚclassmethodÚargparseÚArgumentParserr   ÚstaticmethodÚ	Namespacer   r   r   r   r   r   	   s    r   )r   Úvllm.benchmarks.throughputr   r   Ú#vllm.entrypoints.cli.benchmark.baser   r   r   r   r   r   Ú<module>   s   