|
@@ -22,12 +22,7 @@ from hivemind.averaging.group_info import GroupInfo
|
|
|
from hivemind.averaging.load_balancing import load_balance_peers
|
|
|
from hivemind.averaging.matchmaking import Matchmaking, MatchmakingException
|
|
|
from hivemind.averaging.partition import DEFAULT_PART_SIZE_BYTES
|
|
|
-from hivemind.compression import (
|
|
|
- CompressionBase,
|
|
|
- CompressionInfo,
|
|
|
- NoCompression,
|
|
|
- deserialize_torch_tensor
|
|
|
-)
|
|
|
+from hivemind.compression import CompressionBase, CompressionInfo, NoCompression, deserialize_torch_tensor
|
|
|
from hivemind.dht import DHT, DHTID
|
|
|
from hivemind.p2p import P2P, P2PContext, P2PHandlerError, PeerID, ServicerBase
|
|
|
from hivemind.p2p.p2p_daemon_bindings.utils import ControlFailure, DispatchFailure
|
|
@@ -40,7 +35,7 @@ from hivemind.utils.asyncio import (
|
|
|
as_aiter,
|
|
|
azip,
|
|
|
enter_asynchronously,
|
|
|
- switch_to_uvloop
|
|
|
+ switch_to_uvloop,
|
|
|
)
|
|
|
from hivemind.utils.grpc import combine_from_streaming, split_for_streaming
|
|
|
from hivemind.utils.serializer import MSGPackSerializer, SerializerBase
|