update test_secede_cancelled_or_resumed_scheduler

dask · Sep 8, 2022 · 2eefe23 · 2eefe23
1 parent 7346976
commit 2eefe23
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/distributed/tests/test_cancelled_state.py b/distributed/tests/test_cancelled_state.py
@@ -1034,7 +1034,7 @@ def f(ev1, ev2, ev3, ev4):
     await ev1.wait()
     ts = a.state.tasks["x"]
     assert ts.state == "executing"
-    assert sum(ws.processing.values()) > 0
+    assert any(ws.processing.values())
 
     x.release()
     await wait_for_state("x", "cancelled", a)
@@ -1050,7 +1050,7 @@ def f(ev1, ev2, ev3, ev4):
 
     # Test that the scheduler receives a delayed {op: long-running}
     assert ws.processing
-    while sum(ws.processing.values()):
+    while any(ws.processing.values()):
         await asyncio.sleep(0.1)
     assert ws.processing