From 427fbcb60206f675240ba38c88abda8083f96acb Mon Sep 17 00:00:00 2001 From: James Lamb Date: Sun, 14 Mar 2021 22:44:29 -0500 Subject: [PATCH] [dask] [ci] fix flaky network-setup test --- tests/python_package_test/test_dask.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/python_package_test/test_dask.py b/tests/python_package_test/test_dask.py index d4b22f675c5f..00079d945c5d 100644 --- 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,