From 8a1d16171106a03151eb780070dc5212d360d7e7 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Yuan-Ting=20Hsieh=20=28=E8=AC=9D=E6=B2=85=E5=BB=B7=29?= Date: Fri, 6 Sep 2024 16:49:56 -0700 Subject: [PATCH] Fix xgb standalone fed (#2924) Co-authored-by: Chester Chen <512707+chesterxgchen@users.noreply.github.com> --- .../advanced/xgboost_secure/train_standalone/train_federated.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/advanced/xgboost_secure/train_standalone/train_federated.py b/examples/advanced/xgboost_secure/train_standalone/train_federated.py index 48845087c4..808e88fa17 100644 --- a/examples/advanced/xgboost_secure/train_standalone/train_federated.py +++ b/examples/advanced/xgboost_secure/train_standalone/train_federated.py @@ -63,7 +63,7 @@ def load_test_data(data_path: str): def run_server(port: int, world_size: int) -> None: - xgboost.federated.run_federated_server(port, world_size) + xgboost.federated.run_federated_server(n_workers=world_size, port=port) def run_worker(port: int, world_size: int, rank: int, args) -> None: