:[g |dZddlZddlZddlZddlmZmZddlmZm Z m Z ddl m Z m Z mZdedefd ZGd d Zy) zTracker for XGBoost collective.N)IntEnumunique)DictOptionalUnion)_LIB _check_call make_jcargsaddrreturnc:tj|dddS)z Get network family from address.Nr)socket getaddrinfo)r s J/var/www/html/bid-api/venv/lib/python3.12/site-packages/xgboost/tracker.py get_familyr s   dD )! ,Q //c eZdZdZeGddeZ ddedee dede ded d f d Z dd Z dd Z ddZ ddeed d fdZd ee ee efffdZy ) RabitTrackeraTracker for the collective used in XGBoost, acting as a coordinator between workers. Parameters .......... sortby: How to sort the workers for rank assignment. The default is host, but users can set the `DMLC_TASK_ID` via RABIT initialization arguments and obtain deterministic rank assignment. Available options are: - host - task timeout : Timeout for constructing the communication group and waiting for the tracker to shutdown when it's instructed to, doesn't apply to communication when tracking is running. The timeout value should take the time of data loading and pre-processing into account, due to potential lazy execution. The :py:meth:`.wait_for` method has a different timeout parameter that can stop the tracker even if the tracker is still being used. A value error is raised when timeout is reached. ceZdZdZdZy)RabitTracker._SortByrrN)__name__ __module__ __qualname__HOSTTASKrr_SortByr.s rr n_workershost_ipportsortbytimeoutr Nc htj}|dvr td| t|t |||d|dk(r|j j n|j jt|}ttj|tj|||_ y)N)hosttaskz-Expecting either 'host' or 'task' for sortby.rabitr%)r%rr!dmlc_communicatorr"r#)ctypesc_void_p ValueErrorrr rrrintr r XGTrackerCreatebyrefhandle)selfrr r!r"r#r/argss r__init__zRabitTracker.__init__3s" ) )LM M   w %(.&(84<<$$dll>O>OL   D((v||F/CDE rcvt|dr-|j}|`ttj|yy)zInternal function for testing.r/N)hasattrr/r r XGTrackerFree)r0r/s rfreezRabitTracker.freeLs2 4 "[[F **62 3 #rc$|jyN)r6r0s r__del__zRabitTracker.__del__Ss  rcfttj|jt y)zStart the tracker. Once started, the client still need to call the :py:meth:`wait_for` method in order to wait for it to finish (think of it as a thread). N)r r XGTrackerRunr/r r9s rstartzRabitTracker.startVs D%%dkk;=ABrcjttj|jt |y)zWait for the tracker to finish all the work and shutdown. When timeout is reached, a value error is raised. By default we don't have timeout since we don't know how long it takes for the model to finish training. )r#N)r r XGTrackerWaitForr/r )r0r#s rwait_forzRabitTracker.wait_for^s" D))$++{77STUrctj}ttj|j tj ||jJtj|j}|S)zGet arguments for workers.) r)c_char_pr r XGTrackerWorkerArgsr/r.valuejsonloads)r0c_envenvs r worker_argszRabitTracker.worker_argsfsW!D,,T[[&,,u:MNO{{&&&jj% r)rr%r)r Nr8)rrr__doc__rrrr,rstrr2r6r:r=r@rrrIrrrrrs8 '  #      24CV VVT#uS#X"67rr)rJr)rErenumrrtypingrrrcorer r r rKr,rrrrrrOs?% ((000S0S0 [[r