[dask] [ci] fix flaky network-setup test (#4071)

microsoft · Mar 15, 2021 · 39c85dd · 39c85dd
1 parent 296b2a2
commit 39c85dd
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tests/python_package_test/test_dask.py b/tests/python_package_test/test_dask.py
@@ -1024,6 +1024,8 @@ def test_network_params_not_required_but_respected_if_given(client, task, output
     if task == 'ranking' and output == 'scipy_csr_matrix':
         pytest.skip('LGBMRanker is not currently tested on sparse matrices')
 
+    client.wait_for_workers(2)
+
     if task == 'ranking':
         _, _, _, _, dX, dy, _, dg = _create_ranking_data(
             output=output,