From 720c509bd6734e3e5f5d777077d95242509d152f 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 | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/tests/python_package_test/test_dask.py b/tests/python_package_test/test_dask.py index d4b22f675c5f..b9b418316e9d 100644 --- a/tests/python_package_test/test_dask.py +++ b/tests/python_package_test/test_dask.py @@ -14,8 +14,8 @@ import lightgbm as lgb -if not platform.startswith('linux'): - pytest.skip('lightgbm.dask is currently supported in Linux environments', allow_module_level=True) +#if not platform.startswith('linux'): +# pytest.skip('lightgbm.dask is currently supported in Linux environments', allow_module_level=True) if not lgb.compat.DASK_INSTALLED: pytest.skip('Dask is not installed', allow_module_level=True) @@ -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,