From 42b98553b721378e1705b0b4b71e18ef02de8b3a Mon Sep 17 00:00:00 2001 From: fjetter Date: Wed, 18 Aug 2021 16:12:31 +0200 Subject: [PATCH] Fix flaky test_worker_reconnects_mid_compute --- distributed/tests/test_worker.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/distributed/tests/test_worker.py b/distributed/tests/test_worker.py index f1895a64895..21920c44da7 100644 --- a/distributed/tests/test_worker.py +++ b/distributed/tests/test_worker.py @@ -2443,7 +2443,7 @@ def fast_on_a(lock): assert "Unexpected worker completed task" in s_logs.getvalue() - while not len(s.tasks[f2.key].who_has) == 2: + while a.address not in {w.address for w in s.tasks[f2.key].who_has}: await asyncio.sleep(0.001) # Ensure that all keys have been properly registered and will also be @@ -2510,7 +2510,7 @@ def fast_on_a(lock): assert "Unexpected worker completed task" in s_logs.getvalue() - while not len(s.tasks[f2.key].who_has) == 2: + while a.address not in {w.address for w in s.tasks[f2.key].who_has}: await asyncio.sleep(0.001) del f1, f2, f3