avoid test failure

microsoft · Mar 10, 2021 · 9442bdf · 9442bdf
1 parent 39359c7
commit 9442bdf
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/tests/python_package_test/test_dask.py b/tests/python_package_test/test_dask.py
@@ -1057,7 +1057,11 @@ def test_network_params_not_required_but_respected_if_given(client, task, output
 
     # model 2 - machines given
     n_workers = len(client.scheduler_info()['workers'])
-    open_ports = [lgb.dask._find_random_open_port() for _ in range(n_workers)]
+    while True:
+        open_ports = [lgb.dask._find_random_open_port() for _ in range(n_workers)]
+        if len(set(open_ports)) == len(open_ports):
+            break
+
     dask_model2 = dask_model_factory(
         n_estimators=5,
         num_leaves=5,