Ensure event loop is created in test_lock_workers

pentschev · Jan 18, 2023 · b035e82 · b035e82
1 parent f0cd366
commit b035e82
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/dask_cuda/tests/test_explicit_comms.py b/dask_cuda/tests/test_explicit_comms.py
@@ -330,6 +330,12 @@ def test_lock_workers():
  except ImportError as e:
  pytest.skip(str(e))
 
+ # Avoids DeprecationWarning from `IOLoop()` due to no event loop getting created
+ # with `asyncio.get_event_loop()` in Python >= 3.10
+ # TODO: Switch to async cluster, simply switching the cluster is raising connection
+ # errors.
+ asyncio.new_event_loop()
+
  with LocalCluster(
  protocol="tcp",
  dashboard_address=None,