o
    `Û·il  ã                   @   s8   d dl Z d dlZd dlmZ e e¡ZG dd„ dƒZdS )é    N)Ú
cli_loggerc                   @   s&   e Zd Zd	dd„Zdd„ Zdd„ ZdS )
ÚLogTimerFc                 C   s   || _ || _d S ©N)Ú_messageÚ_show_status)ÚselfÚmessageÚshow_status© r
   úW/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/autoscaler/_private/log_timer.pyÚ__init__
   s   
zLogTimer.__init__c                 C   s   t j  ¡ | _d S r   )ÚdatetimeÚutcnowÚ_start_time)r   r
   r
   r   Ú	__enter__   s   zLogTimer.__enter__c              
   G   sb   t jdkrd S tj ¡ | j }d}| jrt|ƒrdnd}t  d | j	|d 
| ¡ d ¡g¡¡ d S )NÚrecordÚ ÚfailedÚ	succeededú z[LogTimer={:.0f}ms]iè  )r   Ú	log_styler   r   r   r   ÚanyÚprintÚjoinr   ÚformatÚtotal_seconds)r   Ú
error_valsÚtdÚstatusr
   r
   r   Ú__exit__   s   
ýÿÿzLogTimer.__exit__N)F)Ú__name__Ú
__module__Ú__qualname__r   r   r   r
   r
   r
   r   r   	   s    
r   )r   ÚloggingÚ"ray.autoscaler._private.cli_loggerr   Ú	getLoggerr    Úloggerr   r
   r
   r
   r   Ú<module>   s
    
