Mark test_worker_reconnects_mid_compute* tests as flaky (#5378)

dask · Oct 5, 2021 · 07d8e4c · 07d8e4c
1 parent 7bf884b
commit 07d8e4c
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/distributed/tests/test_worker.py b/distributed/tests/test_worker.py
@@ -2444,6 +2444,7 @@ async def test_hold_on_to_replicas(c, s, *workers):
         await asyncio.sleep(0.01)
 
 
+@pytest.mark.flaky(reruns=10, reruns_delay=5)
 @gen_cluster(client=True)
 async def test_worker_reconnects_mid_compute(c, s, a, b):
     """Ensure that, if a worker disconnects while computing a result, the scheduler will
@@ -2512,6 +2513,7 @@ def fast_on_a(lock):
         await asyncio.sleep(0.001)
 
 
+@pytest.mark.flaky(reruns=10, reruns_delay=5)
 @gen_cluster(client=True)
 async def test_worker_reconnects_mid_compute_multiple_states_on_scheduler(c, s, a, b):
     """