o
    `۷i'                     @   s<   d dl Z d dlmZ dd ZdefddZdefd	d
ZdS )    N)RayDeprecationWarningc                    s    fdd}|S )Nc                    s    j | _ | S )N)__doc__)func	copy_func E/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/train/utils.pywrapped   s   z_copy_doc.<locals>.wrappedr   )r   r	   r   r   r   	_copy_doc   s   r
   messagec                 C   s   t j| tdd d S )N   )
stacklevel)warningswarnr   )r   r   r   r   _log_deprecation_warning   s
   
r   returnc                  C   s6   ddl m}  |  rddlm} | S ddlm} | S )Nr   )is_v2_enabled)_in_ray_train_worker) ray.train.v2._internal.constantsr   ray.train.v2._internal.utilr   ray.train._internal.session)r   _in_ray_train_v2_worker_in_ray_train_v1_workerr   r   r   r      s   r   )r   ray.util.annotationsr   r
   strr   boolr   r   r   r   r   <module>   s
    