From c95aafd981ba24774fab4e5adccb864449cacd52 Mon Sep 17 00:00:00 2001 From: idorrington92 Date: Thu, 20 Apr 2023 16:26:46 +0100 Subject: [PATCH] Changing test to make it work in python 3.8 --- distributed/deploy/tests/test_cluster.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/distributed/deploy/tests/test_cluster.py b/distributed/deploy/tests/test_cluster.py index 73ed0cfd897..5533916072e 100644 --- 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