From 9c8ff86ecb5f26c2e8da4510c8a320bb396ead1c Mon Sep 17 00:00:00 2001 From: Gabe Joseph Date: Fri, 28 Oct 2022 05:45:31 -0400 Subject: [PATCH] Fix flaky test_include_communication_in_occupancy (#7212) --- distributed/tests/test_scheduler.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/distributed/tests/test_scheduler.py b/distributed/tests/test_scheduler.py index 3c353901b74..3ed53391dd9 100644 --- a/distributed/tests/test_scheduler.py +++ b/distributed/tests/test_scheduler.py @@ -1472,7 +1472,12 @@ async def test_balance_many_workers_2(c, s, *workers): assert {len(w.has_what) for w in s.workers.values()} == {3} -@gen_cluster(client=True) +@gen_cluster( + client=True, + worker_kwargs={ + "heartbeat_interval": "10s", # prevent worker from updating executing task durations + }, +) async def test_include_communication_in_occupancy(c, s, a, b): x = c.submit(operator.mul, b"0", int(s.bandwidth) * 2, workers=a.address) y = c.submit(operator.mul, b"1", int(s.bandwidth * 3), workers=b.address)