Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

[dask] Use DMLC_TASK_ID. (#5415)

Browse Source
This commit is contained in:
Jiaming Yuan 2020-03-15 16:47:03 +08:00 committed by GitHub
parent fc88105620
commit 668e432e2d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
python-package/xgboost/dask.py
View File

@ -69,6 +69,9 @@ class RabitContext:
'''A context controling rabit initialization and finalization.''' '''A context controling rabit initialization and finalization.'''
def __init__(self, args): def __init__(self, args):
self.args = args self.args = args
worker = distributed_get_worker()
self.args.append(
('DMLC_TASK_ID=[xgboost.dask]:' + str(worker.address)).encode())
def __enter__(self): def __enter__(self):
rabit.init(self.args) rabit.init(self.args)