^j dZddlZddlmZddlmZmZmZddlm Z m Z m Z m Z ddl mZGdd eZe dddd d d d ededeedeedeedededeeeeffdZy)z4XGBoost Experimental Federated Learning related API.N)Thread)AnyDictOptional)_LIB _check_call_deprecate_positional_args make_jcargs) RabitTrackerc`eZdZdZeddddddedededeed eed eed ed dfd Z y)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 tj}t||d||||t|} t t j | tj|||_y)N federated)rrdmlc_communicatorfederated_securerrrr) ctypesc_void_pr intr rXGTrackerCreatebyrefhandle) selfrrrrrrrr!argss \/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/xgboost/federated.py__init__zFederatedTracker.__init__%s]")#+--L   D((v||F/CDE ) __name__ __module__ __qualname____doc__r rboolrstrr%r&r$rr s2 *.*.*.   "# #3-#3-  r&rTr)rrrblockingrrrrrrr.rrc :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..Vs  Ds)rrrrrrrN)targetT)allrstartwait_forrdaemonupdate worker_args) rrrrrr.rr#rtrackerthreads r$run_federated_serverr=@s*(3D $&68HIF '))G MMO 7++ ,FFM LLNKK##%& Kr&)r*r threadingrtypingrrrcorerr r r r;r rrr,r+r=r-r&r$rAs: &&LL!2|2j &*&*&*,, ,c] , sm , sm ,,,d38n,,r&