Alexander Borzunov 4 anos atrás
pai
commit
c26929c397
1 arquivos alterados com 2 adições e 2 exclusões
  1. 2 2
      tests/test_util_modules.py

+ 2 - 2
tests/test_util_modules.py

@@ -17,13 +17,13 @@ from hivemind.utils.asyncio import (
     achain,
     achain,
     aenumerate,
     aenumerate,
     afirst,
     afirst,
-    as_aiter,
+    aiter_with_timeout,
     amap_in_executor,
     amap_in_executor,
     anext,
     anext,
+    as_aiter,
     asingle,
     asingle,
     azip,
     azip,
     cancel_and_wait,
     cancel_and_wait,
-    aiter_with_timeout,
 )
 )
 from hivemind.utils.compression import deserialize_torch_tensor, serialize_torch_tensor
 from hivemind.utils.compression import deserialize_torch_tensor, serialize_torch_tensor
 from hivemind.utils.mpfuture import InvalidStateError
 from hivemind.utils.mpfuture import InvalidStateError