From 7a8fb609d25159f4e8e1c7dadc361bca40fe28da Mon Sep 17 00:00:00 2001 From: David Farrington Date: Tue, 25 Jun 2024 12:42:07 +0100 Subject: [PATCH] Name arguments to rabit tracker to maintain compatability between xgboost 2.0.3 & 2.1.0 --- xgboost_ray/main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/xgboost_ray/main.py b/xgboost_ray/main.py index 0fbe2fd..d5dd459 100644 --- a/xgboost_ray/main.py +++ b/xgboost_ray/main.py @@ -272,7 +272,8 @@ def _start_rabit_tracker(num_workers: int): env = {"DMLC_NUM_WORKER": num_workers} - rabit_tracker = _RabitTracker(host, num_workers) + # Name arguments, to ensure compatability with newer xgboost versions + rabit_tracker = _RabitTracker(host_ip=host, n_workers=num_workers) # Get tracker Host + IP env.update(rabit_tracker.worker_envs())