o
    $i                     @   s8   d dl mZ d dlmZ d dlmZ G dd deZdS )    )contextmanager)get_train_fn_utils)XGBoostConfigc                       s    e Zd Ze fddZ  ZS )r   c                    s    t t| j t fdd}|S )Nc                   3   sH    t   r   d V  W d    d S 1 sw   Y  d S d V  d S )N)r   is_distributed distributed_contextr   X/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/ray/train/v2/xgboost/config.py collective_communication_context   s   
"
zJXGBoostConfig.train_func_context.<locals>.collective_communication_context)superr   train_func_contextr   )selfr
   	__class__r   r	   r      s   z XGBoostConfig.train_func_context)__name__
__module____qualname__propertyr   __classcell__r   r   r   r	   r      s    r   N)
contextlibr   /ray.train.v2._internal.execution.train_fn_utilsr   ray.train.xgboost.configr   XGBoostConfigV1r   r   r   r	   <module>   s    