o
    ?߱i                     @   s"   d dl mZ eG dd dZdS )    )	dataclassc                   @   sJ   e Zd ZU dZdZeed< dZeed< edefddZ	de
fdd	Zd
S )BenchmarkTimeszi
    Class used to store times computed during tokenizer benchmarking.
    All times are in seconds.
    g        model_invocationtotalreturnc                 C   s   | j | j S )N)r   r   self r	   d/data/cameron/vidgen/cosmos-predict2.5/cosmos_predict2/_src/predict2/utils/tokenizer_benchmarking.pyoverhead   s   zBenchmarkTimes.overheadc                 C   s   d| j  d| j d| j dS )Nz BenchmarkTimes(model_invocation=z, overhead=z, total=))r   r   r   r   r	   r	   r
   __repr__"   s   zBenchmarkTimes.__repr__N)__name__
__module____qualname____doc__r   float__annotations__r   propertyr   strr   r	   r	   r	   r
   r      s   
 r   N)dataclassesr   r   r	   r	   r	   r
   <module>   s   