Skip to content

Commit

Permalink
Properly await async method client.wait_for_workers (#8558) (#8567)
Browse files Browse the repository at this point in the history
* Properly await async method client.wait_for_workers

* ignore mypy error.

Co-authored-by: jiamingy <jm.yuan@outlook.com>

Co-authored-by: Matthew Rocklin <mrocklin@gmail.com>
  • Loading branch information
trivialfis and mrocklin committed Dec 7, 2022
1 parent 59c54e3 commit a980e10
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion python-package/xgboost/dask.py
Expand Up @@ -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
)
Expand Down

0 comments on commit a980e10

Please sign in to comment.