^j~ (dZddlZddlZddlmZddlmZmZm Z m Z ddl Z ddl Z ddl mZddl mZddlmZej&d Zd eeefd d d efdZde eeefded e e eeffdZed efdZed efdZed efdZy)z)Utilities for the XGBoost Dask interface.N)cache)AnyDictOptionalTuple)Version)parse)Configz[xgboost.dask] local_paramworkerzdistributed.Workerreturnc|jj}d}dD]F}|j|d|j|||k7s,tj d||}n|dk(s||}|S)zIGet the number of threads from a worker and the user-supplied parameters.N)nthreadn_jobsz-Overriding `nthreads` defined in dask worker.r)statenthreadsgetLOGGERinfo)r r dwnt n_threadsps ]/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/xgboost/dask/utils.py get_n_threadsrs} << DI " ??1d # /KOOAt4LPT4T KKG H#AI   A~* dconfigcoll_cfgcdg}d}d}|8|D]2}||vrtd|tjd|dt4ni}|j dd}|'|j dr|j dr|d d }|# tjj|\}}| t}|j9|+|j|k7rtd |d |jd|j}|jM|dk7r<|:|jdk7r+||jk7rtd |d |jd|j}||fS#t$rYwxYw)a Get the tracker address from the optional user configuration. Parameters ---------- dconfig : Dask global configuration. coll_cfg : Collective configuration. Returns ------- The IP address along with the port number. scheduler_addressNrzUnknown configuration: zeUse `coll_cfg` instead of the Dask global configuration store for the XGBoost tracker configuration: .[]z\Conflicting host IP addresses from the dask configuration and the collective configuration: z v.s. zPConflicting ports from the dask configuration and the collective configuration: ) ValueErrorwarningswarn FutureWarningr startswithendswith distributedcommget_address_host_portr tracker_host_ip tracker_port)rr valid_confighost_ipportks rget_address_from_userr5!s&((LG D A $ #:1#!>?? MM??@cD   kk-t4Gw11#67;K;KC;P!B- ',,BB7KMGT8+  8#;#;w#F--4IVHos ?mK8 88rc.ttdk\S)Nz2025.3.0r=r:rr_DASK_2025_3_0r@ts ?mJ7 77r)__doc__loggingr' functoolsrfcachetypingrrrrr8r,packaging.versionrr r7 collectiver getLoggerrstrintrr5r;boolr>r@r:rrrLs/%-- %4   + , tCH~ 7K PS F d38n %F17F 8C=# FR+w++999888r