Handle stream is None case in TCP comm finalizer (#4631)

dask · Mar 25, 2021 · 43859e2 · 43859e2
1 parent 3b8b97e
commit 43859e2
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/distributed/comm/tcp.py b/distributed/comm/tcp.py
@@ -169,8 +169,9 @@ def _read_extra(self):
 
     def _get_finalizer(self):
         def finalize(stream=self.stream, r=repr(self)):
-            if not stream.closed():
-                logger.warning("Closing dangling stream in %s" % (r,))
+            # stream is None if a StreamClosedError is raised during interpreter shutdown
+            if stream is not None and not stream.closed():
+                logger.warning(f"Closing dangling stream in {r}")
                 stream.close()
 
         return finalize