Fix _update_scheduler_info hanging failed tests (#7225)

dask · Nov 1, 2022 · 5a14053 · 5a14053
1 parent a573da0
commit 5a14053
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/distributed/client.py b/distributed/client.py
@@ -1361,7 +1361,7 @@ async def _ensure_connected(self, timeout=None):
         logger.debug("Started scheduling coroutines. Synchronized")
 
     async def _update_scheduler_info(self):
-        if self.status not in ("running", "connecting"):
+        if self.status not in ("running", "connecting") or self.scheduler is None:
             return
         try:
             self._scheduler_identity = SchedulerInfo(await self.scheduler.identity())