From d731381b90cad042a5ab33487251fb3e8d542f7b Mon Sep 17 00:00:00 2001 From: crusaderky Date: Fri, 15 Oct 2021 09:41:14 +0100 Subject: [PATCH] f --- distributed/worker.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/distributed/worker.py b/distributed/worker.py index 02c7cfb5642..ab82087911b 100644 --- a/distributed/worker.py +++ b/distributed/worker.py @@ -911,9 +911,9 @@ def executor(self): def status(self, value): """Override Server.status to notify the Scheduler of status changes""" ServerNode.status.__set__(self, value) - self._send_worker_status() + self._send_worker_status_change() - def _send_worker_status(self) -> None: + def _send_worker_status_change(self) -> None: if ( self.batched_stream and self.batched_stream.comm @@ -923,7 +923,7 @@ def _send_worker_status(self) -> None: {"op": "worker-status-change", "status": self._status.name} ) elif self._status != Status.closed: - self.loop.call_later(0.05, self._send_worker_status) + self.loop.call_later(0.05, self._send_worker_status_change) async def get_metrics(self): out = dict( @@ -3696,7 +3696,11 @@ def get_worker() -> Worker: return thread_state.execution_state["worker"] except AttributeError: try: - return first(w for w in Worker._instances if w.status in RUNNING) + return first( + w + for w in Worker._instances + if w.status in (Status.running, Status.paused) + ) except StopIteration: raise ValueError("No workers found")