Change address.

dmlc · Oct 8, 2020 · c027a37 · c027a37
1 parent 37f3562
commit c027a37
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask.py
@@ -608,8 +608,8 @@ def _dmatrix_from_worker_map(is_quantile, **kwargs):
 
 async def _get_rabit_args(worker_map, client: Client):
     '''Get rabit context arguments from data distribution in DaskDMatrix.'''
-    host = distributed_comm.get_address_host(client.scheduler.address)
     tracker_worker = [worker_map[0]]
+    host = distributed_comm.get_address_host(tracker_worker[0])
     n_workers = len(worker_map)
     envs = await client.submit(_start_tracker, host.strip('/:'),
                                n_workers, workers=tracker_worker)