From 74a1fde174087250e696c35c59f26129fb238d61 Mon Sep 17 00:00:00 2001 From: Matthew Rocklin Date: Wed, 7 Dec 2022 07:49:30 -0600 Subject: [PATCH] Properly await async method client.wait_for_workers (#8558) * Properly await async method client.wait_for_workers * ignore mypy error. Co-authored-by: jiamingy --- python-package/xgboost/dask.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python-package/xgboost/dask.py b/python-package/xgboost/dask.py index 7d21ff9fcb48..69d1832f1506 100644 --- a/python-package/xgboost/dask.py +++ b/python-package/xgboost/dask.py @@ -853,7 +853,7 @@ async def _get_rabit_args( sched_addr = None # make sure all workers are online so that we can obtain reliable scheduler_info - client.wait_for_workers(n_workers) + await client.wait_for_workers(n_workers) # type: ignore env = await client.run_on_scheduler( _start_tracker, n_workers, sched_addr, user_addr )