[dask] Use DMLC_TASK_ID. (#5415)

dmlc · Mar 15, 2020 · 668e432 · 668e432
1 parent fc88105
commit 668e432
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask.py
@@ -69,6 +69,9 @@ class RabitContext:
     '''A context controling rabit initialization and finalization.'''
     def __init__(self, args):
         self.args = args
+        worker = distributed_get_worker()
+        self.args.append(
+            ('DMLC_TASK_ID=[xgboost.dask]:' + str(worker.address)).encode())
 
     def __enter__(self):
         rabit.init(self.args)