:Dj dZddlZddlmZddlmZmZmZddlm Z m Z m Z ddl m Z Gdd e Z dd ed ed eed eedeedededeeeeffdZy)z4XGBoost Experimental Federated Learning related API.N)Thread)AnyDictOptional)_LIB _check_call make_jcargs) RabitTrackercTeZdZdZ d dedededeedeedeed ed dfd Zy) FederatedTrackeraTracker for federated training. Parameters ---------- n_workers : The number of federated workers. port : The port to listen on. secure : Whether this is a secure instance. If True, then the following arguments for SSL must be provided. server_key_path : Path to the server private key file. server_cert_path : Path to the server certificate file. client_cert_path : Path to the client certificate file. N n_workersportsecureserver_key_pathserver_cert_pathclient_cert_pathtimeoutreturnc tj}t||d||||t|} t t j | tj|||_y)N federated)rrdmlc_communicatorfederated_securerrrr) ctypesc_void_pr intr rXGTrackerCreatebyrefhandle) selfrrrrrrrrargss UC:\Crop_Prediction\Backend\crop-ai-system\venv\Lib\site-packages\xgboost/federated.py__init__zFederatedTracker.__init__%s]")#+--L   D((v||F/CDE )NNN,) __name__ __module__ __qualname____doc__rboolrstrr#r$r"r r st<*.*.*.  "#  #3- #3- r$r rrrrrblockingrrc :d|i}td|||fD}t|||||||} | j|r| jyt | j} d| _| j|j | j|S)a|See :py:class:`~xgboost.federated.FederatedTracker` for more info. Parameters ---------- blocking : Block the server until the training is finished. If set to False, the function launches an additional thread and returns the worker arguments. The default is True and a higher level framework is responsible for setting worker parameters. rc3$K|]}|du yw)Nr,).0paths r" z'run_federated_server..Rs  Ds)rrrrrrrN)targetT)allr startwait_forrdaemonupdate worker_args) rrrrrr-rr!rtrackerthreads r"run_federated_serverr<>s&(3D $&68HIF '))G MMO 7++ ,FFM LLNKK##%& Kr$)NNNTr%)r)r threadingrtypingrrrcorerr r r:r r rr+r*r<r,r$r"r@s: &&00!0|0l&*&*&*++ +c]+sm + sm +  ++d38n+r$