o ٭j @sdZddlZddlZddlmZddlmZmZm Z m Z ddl Z ddl Z ddl mZddl mZddlmZed Zd eeefd d d efddZde eeefded e e eeffddZed efddZed efddZed efddZdS)z)Utilities for the XGBoost Dask interface.N)cache)AnyDictOptionalTuple)Version)parse)Configz[xgboost.dask] local_paramworkerzdistributed.WorkerreturncCstt|dr |jjn|j}d}dD]}||ddur-||||kr-td||}nq|dks6|dur8|}|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_threadsprO/var/www/html/CropPilot/venv/lib/python3.10/site-packages/xgboost/dask/utils.py get_n_threadss  rdconfigcoll_cfgcCsPdg}d}d}|dur&|D]}||vrtd|td|dtq ni}|dd}|durB|drB|drB|d d }|durZz tj |\}}Wn tyYYnw|durat }|j dur~|dur{|j |kr{td |d |j d|j }|j dur|dkr|dur|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_ipportkrrrget_address_from_user!sd       r3cCs ttjS)N) parse_versiondask __version__rrrr _DASK_VERSIONjs r7cCttdkS)Nz 2024.12.1r7r4rrrr_DASK_2024_12_1or:cCr8)Nz2025.3.0r9rrrr_DASK_2025_3_0tr;r<)__doc__loggingr% functoolsrfcachetypingrrrrr5r*packaging.versionrrr4 collectiver getLoggerrstrintrr3r7boolr:r<rrrrs2      I