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

2021-03-29 03:47:56 -05:00 · 2021-03-29 03:47:56 -05:00 · f01af43eb0
commit f01af43eb0
parent a59c7323b4
1 changed files with 1 additions and 0 deletions
--- a/python-package/xgboost/dask.py
+++ b/python-package/xgboost/dask.py
@ -907,6 +907,7 @@ async def _train_async(
                evals_per_worker,
                pure=False,
                workers=[worker_addr],
+                allow_other_workers=False
            )
            futures.append(f)