[dask] [ci] fix flaky network-setup test

microsoft · Mar 15, 2021 · 720c509 · 720c509
1 parent b044070
commit 720c509
Showing 1 changed file with 4 additions and 2 deletions.
diff --git 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,