use gen_cluster timeout

gjoseph92 · Jun 21, 2021 · c3539c2 · c3539c2
1 parent 97f9d60
commit c3539c2
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/distributed/tests/test_actor.py b/distributed/tests/test_actor.py
@@ -466,7 +466,7 @@ def f(block, ps=None):
 
 
 @pytest.mark.flaky(reruns=10, reruns_delay=5)
-@gen_cluster(client=True)
+@gen_cluster(client=True, timeout=10)
 async def test_compute(c, s, a, b):
     @dask.delayed
     def f(n, counter):
@@ -485,10 +485,8 @@ def check(counter, blanks):
     result = await c.compute(final, actors=counter)
     assert result == 0 + 1 + 2 + 3 + 4
 
-    start = time()
     while a.data or b.data:
         await asyncio.sleep(0.01)
-        assert time() < start + 30
 
 
 def test_compute_sync(client):