o ٭j @sdZddlZddlmZddlmZmZmZddlm Z m Z m Z m Z ddl mZGdd d eZe dddd d d d ededeedeedeedededeeeeffddZdS)z4XGBoost Experimental Federated Learning related API.N)Thread)AnyDictOptional)_LIB _check_call_deprecate_positional_args make_jcargs) RabitTrackerc@sVeZdZdZeddddddedededeed eed eed ed dfd dZ dS)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,)server_key_pathserver_cert_pathclient_cert_pathtimeout n_workersportsecurerrrrreturnc CsDt}t||d||||t|d} tt| t|||_dS)N federated)rrdmlc_communicatorfederated_securerrrr) ctypesc_void_pr intrrXGTrackerCreatebyrefhandle) selfrrrrrrrrargsr!N/var/www/html/CropPilot/venv/lib/python3.10/site-packages/xgboost/federated.py__init__%s  zFederatedTracker.__init__) __name__ __module__ __qualname____doc__r rboolrstrr#r!r!r!r"r s0  r Tr )rrrblockingrrrrrrr*rrc Cszd|i}tdd|||fD}t|||||||d} | |r'| dSt| jd} d| _| || |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. rcss|]}|duVqdS)Nr!).0pathr!r!r" Vs  z'run_federated_server..)rrrrrrrN)targetT)allr startwait_forrdaemonupdate worker_args) rrrrrr*rr rtrackerthreadr!r!r"run_federated_server@s,  r7)r'r threadingrtypingrrrcorerrr r r5r r rr)r(r7r!r!r!r"s<  5