diff --git a/distributed/tests/test_worker.py b/distributed/tests/test_worker.py index c4481fb04fb..fe101155fe7 100644 --- 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): """