From 3fae185d8e5463947b0a0a2206484a13f581f191 Mon Sep 17 00:00:00 2001 From: Gabe Joseph Date: Thu, 27 Oct 2022 16:00:44 -0600 Subject: [PATCH] Fix flaky test_include_communication_in_occupancy --- 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)