don't heartbeat when closing

TODO make this a separate PR and add tests. Just want to see if it helps CI.
gjoseph92 · Jun 8, 2022 · 7d90e2a · 7d90e2a
1 parent d037f37
commit 7d90e2a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/distributed/worker.py b/distributed/worker.py
@@ -1215,6 +1215,9 @@ async def heartbeat(self):
         if self.heartbeat_active:
             logger.debug("Heartbeat skipped: channel busy")
             return
+        if self.status in {Status.closing, Status.closed, Status.failed}:
+            logger.debug(f"Heartbeat skipped: {self.status=}")
+            return
         self.heartbeat_active = True
         logger.debug("Heartbeat: %s", self.address)
         try: