Fix flaky test_include_communication_in_occupancy (#7212)

dask · Oct 28, 2022 · 9c8ff86 · 9c8ff86
1 parent 581b603
commit 9c8ff86
Showing 1 changed file with 6 additions and 1 deletion.
diff --git 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)