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

[dask] disable work stealing explicitly for training tasks (#6794)

Browse Source
This commit is contained in:
James Lamb 2021-03-29 03:47:56 -05:00 committed by GitHub
parent a59c7323b4
commit f01af43eb0
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -907,6 +907,7 @@ async def _train_async(
evals_per_worker, evals_per_worker,
pure=False, pure=False,
workers=[worker_addr], workers=[worker_addr],
allow_other_workers=False
) )
futures.append(f) futures.append(f)