black, isort

learning-at-home · Aug 24, 2023 · 4cf9cdf · 4cf9cdf
1 parent 6df4e32
commit 4cf9cdf
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/hivemind/utils/compat.py b/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)

diff --git a/tests/test_util_modules.py b/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