Changing test to make it work in python 3.8

dask · Apr 20, 2023 · c95aafd · c95aafd
1 parent d277ea6
commit c95aafd
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/distributed/deploy/tests/test_cluster.py b/distributed/deploy/tests/test_cluster.py
@@ -40,11 +40,11 @@ async def test_logs_deprecated():
 
 
 @gen_test()
-async def test_cluster_wait_for_worker(loop):
-    with LocalCluster(n_workers=2, loop=loop) as cluster:
+async def test_cluster_wait_for_worker():
+    async with LocalCluster(n_workers=2, asynchronous=True) as cluster:
         assert len(cluster.scheduler.workers) == 2
         cluster.scale(4)
-        cluster.wait_for_workers(4)
+        await cluster.wait_for_workers(4)
         assert all(
             [
                 worker["status"] == Status.running.name