black, isort

Author: borzunov

hivemind/utils/compat.py

@@ -14,9 +14,8 @@ def safe_recv(pipe: mp.connection.Connection) -> Any:
         try:
             return pipe.recv()
         except Exception as e:
-            if (
-                (isinstance(e, BlockingIOError) and str(e) == "[Errno 35] Resource temporarily unavailable") or
-                (isinstance(e, EOFError) and str(e) == "Ran out of input")
+            if (isinstance(e, BlockingIOError) and str(e) == "[Errno 35] Resource temporarily unavailable") or (
+                isinstance(e, EOFError) and str(e) == "Ran out of input"
             ):
                 logger.warning(repr(e))
                 time.sleep(0)

tests/test_util_modules.py

@@ -13,7 +13,6 @@
 from hivemind.compression import deserialize_torch_tensor, serialize_torch_tensor
 from hivemind.proto.runtime_pb2 import CompressionType
 from hivemind.utils import BatchTensorDescriptor, DHTExpiration, HeapEntry, MSGPackSerializer, ValueWithExpiration
-from hivemind.utils.compat import safe_recv
 from hivemind.utils.asyncio import (
     achain,
     aenumerate,
@@ -27,6 +26,7 @@
     cancel_and_wait,
     enter_asynchronously,
 )
+from hivemind.utils.compat import safe_recv
 from hivemind.utils.mpfuture import InvalidStateError
 from hivemind.utils.performance_ema import PerformanceEMA