o `۷i @sUddlZddlZddlmZddlmZmZmZddlZddl m Z ddl m Z ddl mZddlmZmZeeZdaeeeefed<eZd eeeffd d Zd ed edefddZeGdddeZGdddeZdS)N) dataclass)AnyDictOptional build_address)BaseWorkerGroup)get_address_and_port)Backend BackendConfig_lightgbm_network_paramsreturncCsPttstdiWdStWdS1s!wYdS)zGReturns the network parameters to enable LightGBM distributed training.a`ray.train.lightgbm.get_network_params` was called outside the context of a `ray.train.lightgbm.LightGBMTrainer`. The current process has no knowledge of the distributed training worker group, so this method will return an empty dict. Please call this within the training loop of a `ray.train.lightgbm.LightGBMTrainer`. If you are in fact calling this within a `LightGBMTrainer`, this is unexpected: please file a bug report to the Ray Team.N)_lightgbm_network_params_lockr loggerwarningcopyrrO/home/ubuntu/vllm_env/lib/python3.10/site-packages/ray/train/lightgbm/config.pyget_network_paramss $r num_machineslocal_listen_portmachinescCsHttdus Jdt|||daWddS1swYdS)Nz0LightGBM network params are already initialized.rrr)rr dictrrrr_set_network_params)s "rc@seZdZdZeddZdS)LightGBMConfiga Configuration for LightGBM distributed data-parallel training setup. See the LightGBM docs for more information on the "network parameters" that Ray Train sets up for you: https://lightgbm.readthedocs.io/en/latest/Parameters.html#network-parameters cCstS)N)_LightGBMBackend)selfrrr backend_clsDszLightGBMConfig.backend_clsN)__name__ __module__ __qualname____doc__propertyrrrrrr;src@seZdZdedefddZdS)r worker_groupbackend_configcs^t}dd|Dddd|DttfddttDdS)NcSsg|]\}}|qSrr).0_portrrr Nsz6_LightGBMBackend.on_training_start..,cSsg|] \}}t||qSrr)r&node_ipr(rrrr)Psc s"g|] }|t|qSr)execute_single_asyncr)r&rankrrportsr$rrr)Ts )executer joinlenraygetrange)rr$r%node_ips_and_portsrr.ron_training_startJs   z"_LightGBMBackend.on_training_startN)rr r!rrr7rrrrrIs r)logging threading dataclassesrtypingrrrr3ray._common.network_utilsr%ray.train._internal.base_worker_grouprray.train._internal.utilsr ray.train.backendr r getLoggerrrr str__annotations__Lockrrintrrrrrrrs.