
    &`i                     F    d dl mZ d dlmZ d dlmZ  G d de          ZdS )    )contextmanager)get_train_fn_utils)XGBoostConfigc                   .     e Zd Ze fd            Z xZS )r   c                 b    t          t          |           j        t          fd            }|S )Nc               3      K   t                                                      r)              5  d V  d d d            d S # 1 swxY w Y   d S d V  d S )N)r   is_distributed)distributed_contexts   o/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/ray/train/v2/xgboost/config.py collective_communication_contextzJXGBoostConfig.train_func_context.<locals>.collective_communication_context   s       "##2244 ((**  EEE                  s   A  AA)superr   train_func_contextr   )selfr   r
   	__class__s     @r   r   z XGBoostConfig.train_func_context   sB    #M488K		 	 	 	 
	 0/    )__name__
__module____qualname__propertyr   __classcell__)r   s   @r   r   r      sB        0 0 0 0 X0 0 0 0 0r   r   N)
contextlibr   /ray.train.v2._internal.execution.train_fn_utilsr   ray.train.xgboost.configr   XGBoostConfigV1 r   r   <module>r      so    % % % % % % N N N N N N E E E E E E0 0 0 0 0O 0 0 0 0 0r   