From f01af43eb04f4806b3e35f6fa3744576cc5c8d42 Mon Sep 17 00:00:00 2001 From: James Lamb Date: Mon, 29 Mar 2021 03:47:56 -0500 Subject: [PATCH] [dask] disable work stealing explicitly for training tasks (#6794) --- python-package/xgboost/dask.py | 1 + 1 file changed, 1 insertion(+) diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask.py index 4bdeb49e5..3082b93e8 100644 --- 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)