^i (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.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)hasattrrnthreadsgetLOGGERinfo)r r dwnt n_threadsps S/var/www/html/hubwallet-dev/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_portr tracker_host_ip tracker_port)rr valid_confighost_ipportks rget_address_from_userr6"s&((LG D A $ #:1#!>?? MM??@cD   kk-t4Gw11#67;K;KC;P!B- ',,BB7KMGT8+  8#;#;w#F--4IVHr;rr_DASK_2025_3_0rAus ?mJ7 77r)__doc__loggingr( functoolsrfcachetypingrrrrr9r-packaging.versionrr r8 collectiver getLoggerrstrintrr6r<boolr?rAr;rrrMs/%-- %4   + , tCH~ 7K PS F d38n %F17F 8C=# FR+w++999888r