Update tracker

mars-project · Mar 24, 2022 · 902bb10 · 902bb10
1 parent 819790c
commit 902bb10
Show file tree

Hide file tree

Showing 2 changed files with 265 additions and 121 deletions.
diff --git a/mars/learn/contrib/xgboost/start_tracker.py b/mars/learn/contrib/xgboost/start_tracker.py
@@ -47,9 +47,9 @@ def execute(cls, ctx, op):
 
         env = {"DMLC_NUM_WORKER": op.n_workers}
         rabit_context = RabitTracker(
-            hostIP=ctx.get_local_host_ip(), nslave=op.n_workers
+            host_ip=ctx.get_local_host_ip(), n_workers=op.n_workers
         )
-        env.update(rabit_context.slave_envs())
+        env.update(rabit_context.worker_envs())
 
         rabit_context.start(op.n_workers)
         thread = Thread(target=rabit_context.join)