[Dask] Asyncio support. (#5862)

2020-07-30 06:23:58 +08:00
parent e4a273e1da
commit fa3715f584
5 changed files with 637 additions and 335 deletions
--- a/tests/python/test_tracker.py
+++ b/tests/python/test_tracker.py
@@ -27,8 +27,10 @@ def run_rabit_ops(client, n_workers):
    from xgboost import rabit

    workers = list(_get_client_workers(client).keys())
-    rabit_args = _get_rabit_args(workers, client)
+    rabit_args = client.sync(_get_rabit_args, workers, client)
    assert not rabit.is_distributed()
+    n_workers_from_dask = len(workers)
+    assert n_workers == n_workers_from_dask

    def local_test(worker_id):
        with RabitContext(rabit_args):