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
   ^/home/ubuntu/veenaModal/venv/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
    
