bi- dZddlZddlZddlmZmZmZmZddlZddl m Z ejdZ dee efdd d efd Zd eee efd e d eee effdZy)z)Utilities for the XGBoost Dask interface.N)AnyDictOptionalTuple)Configz[xgboost.dask] local_paramworkerzdistributed.Workerreturnct|dr|jjn |j}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.stateN)nthreadn_jobsz-Overriding `nthreads` defined in dask worker.r)hasattrr nthreadsgetLOGGERinfo)r r dwnt n_threadsps M/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/xgboost/dask/utils.py get_n_threadsrs%,FG$<6<< &//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_portrtracker_host_ip tracker_port)rr valid_confighost_ipportks rget_address_from_userr3s&((LG D A $ #:1#!>?? MM??@cD   kk-t4Gw11#67;K;KC;P!B- ',,BB7KMGT8+  8#;#;w#F--4IVHr<s/--   + , tCH~ 7K PS F d38n %F17F 8C=# Fr