rii VdZddlZddlZddlmZddlmZmZm Z m Z ddl Z ddl Z ddl mZddl mZddlmZejd Zd eeefd d d efdZde eeefded e e eeffdZed efdZed efdZed efdZdS)z)Utilities for the XGBoost Dask interface.N)cache)AnyDictOptionalTuple)Version)parse)Configz[xgboost.dask] local_paramworkerzdistributed.Workerreturnct|dr |jjn|j}d}dD]V}||d>||||kr$td||}nW|dks||}|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 X/var/www/html/bet.cuttalo.com/ml/venv/lib/python3.11/site-packages/xgboost/dask/utils.py get_n_threadsrs%,FG$<$< Q6< &/DI " ??1d # # /KOOAt4L4LPT4T4T KKG H H H#AI EA~~* dconfigcoll_cfgcdg}d}d}|:|D]6}||vrtd|tjd|dt7ni}|dd}|4|dr|dr |d d }|4 tj |\}}n#t$rYnwxYw|t}|j /|&|j |krtd |d |j d|j }|j @|dkr3|1|j dkr&||j krtd |d |j d|j }||fS)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!>!>??? MD?@DDD      kk-t44Gw11#667;K;KC;P;P!B$- ',BB7KKMGTT    D 88+  8#;w#F#FX-4XXr;rr_DASK_2025_3_0rAus ??mJ77 77r)__doc__loggingr( functoolsrfcachetypingrrrrr9r-packaging.versionrr r8 collectiver getLoggerrstrintrr6r<boolr?rAr;rrrMs//%%%%%%------------ %%%%%%444444  + , , tCH~ 7K PS    F d38n %F17F 8C=# FFFFR+w++++999998888888r