瀏覽代碼

more general error

justheuristic 4 年之前
父節點
當前提交
cc3f28655a
共有 2 個文件被更改,包括 2 次插入2 次删除
  1. 1 1
      hivemind/averaging/averager.py
  2. 1 1
      hivemind/dht/__init__.py

+ 1 - 1
hivemind/averaging/averager.py

@@ -243,7 +243,7 @@ class DecentralizedAverager(mp.Process, ServicerBase):
                 while True:
                 while True:
                     try:
                     try:
                         method, args, kwargs = await loop.run_in_executor(pipe_awaiter, self._inner_pipe.recv)
                         method, args, kwargs = await loop.run_in_executor(pipe_awaiter, self._inner_pipe.recv)
-                    except (OSError, BrokenPipeError) as e:
+                    except (OSError, ConnectionError) as e:
                         logger.exception(e)
                         logger.exception(e)
                         await asyncio.sleep(self._matchmaking.request_timeout)
                         await asyncio.sleep(self._matchmaking.request_timeout)
                         continue
                         continue

+ 1 - 1
hivemind/dht/__init__.py

@@ -121,7 +121,7 @@ class DHT(mp.Process):
                 while True:
                 while True:
                     try:
                     try:
                         method, args, kwargs = await loop.run_in_executor(pipe_awaiter, self._inner_pipe.recv)
                         method, args, kwargs = await loop.run_in_executor(pipe_awaiter, self._inner_pipe.recv)
-                    except (OSError, BrokenPipeError) as e:
+                    except (OSError, ConnectionError) as e:
                         logger.exception(e)
                         logger.exception(e)
                         await asyncio.sleep(self._node.protocol.wait_timeout)
                         await asyncio.sleep(self._node.protocol.wait_timeout)
                         continue
                         continue