Merge branch 'AMM/avoid_paused' into AMM/RetireWorker

dask · Oct 15, 2021 · 5d565e5 · 5d565e5
2 parents 7ea9786 + 9b04db9
commit 5d565e5
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/distributed/scheduler.py b/distributed/scheduler.py
@@ -5458,6 +5458,10 @@ def handle_worker_status_change(self, status: str, worker: str) -> None:
         if not ws:
             return
         prev_status = ws._status
+        ws._status = Status.lookup[status]  # type: ignore
+        if ws._status == prev_status:
+            return
+
         self.log_event(
             ws._address,
             {
@@ -5466,9 +5470,6 @@ def handle_worker_status_change(self, status: str, worker: str) -> None:
                 "status": status,
             },
         )
-        ws._status = Status.lookup[status]  # type: ignore
-        if ws._status == prev_status:
-            return
 
         if ws._status == Status.running:
             parent._running.add(ws)