Bladeren bron

fix test warning

Denis Mazur 4 jaren geleden
bovenliggende
commit
1e96285738
2 gewijzigde bestanden met toevoegingen van 4 en 1 verwijderingen
  1. 0 0
      hivemind/moe/server/connection_handler_p2p.py
  2. 4 1
      hivemind/p2p/p2p_daemon_bindings/control.py

+ 0 - 0
hivemind/moe/server/connection_handler_p2p.py


+ 4 - 1
hivemind/p2p/p2p_daemon_bindings/control.py

@@ -147,7 +147,10 @@ class ControlClient:
     async def _read_from_persistent_conn(self, reader: asyncio.StreamReader):
         while True:
             resp = p2pd_pb.PersistentConnectionResponse()
-            await read_pbmsg_safe(reader, resp)
+            try:
+                await read_pbmsg_safe(reader, resp)
+            except asyncio.IncompleteReadError:
+                break
 
             call_id = UUID(bytes=resp.callId)