Minor

hendrikmakait · hendrikmakait · commit 256492446c71 · 2022-05-18T16:36:00.000+02:00
diff --git a/distributed/core.py b/distributed/core.py
@@ -685,7 +685,7 @@ async def close(self):
                 break
         await self.rpc.close()
         await asyncio.gather(
-            *(comm.close() for comm in self._comms.values())
+            *(comm.close() for comm in self._comms)
         )  # then forcefully close
         for cb in self._ongoing_coroutines:
             cb.cancel()
