Fix imports

adap · Feb 23, 2024 · 8b9c854 · 8b9c854
1 parent fa15c5f
commit 8b9c854
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 8 deletions.
diff --git a/src/py/flwr/client/app.py b/src/py/flwr/client/app.py
@@ -25,7 +25,7 @@
 from flwr.client.client import Client
 from flwr.client.clientapp import ClientApp
 from flwr.client.typing import ClientFn
-from flwr.common import GRPC_MAX_MESSAGE_LENGTH, EventType, Message, event
+from flwr.common import GRPC_MAX_MESSAGE_LENGTH, EventType, Message, RecordSet, event
 from flwr.common.address import parse_address
 from flwr.common.constant import (
     MISSING_EXTRA_REST,
@@ -35,13 +35,8 @@
     TRANSPORT_TYPES,
 )
 from flwr.common.logger import log, warn_deprecated_feature, warn_experimental_feature
-
-
-from flwr.common.message import Message
-from flwr.common.recordset import RecordSet
 from flwr.common.serde import message_to_taskres
 
-
 from .clientapp import load_client_app
 from .grpc_client.connection import grpc_connection
 from .grpc_rere_client.connection import grpc_request_response

diff --git a/src/py/flwr/server/compat/driver_client_proxy_test.py b/src/py/flwr/server/compat/driver_client_proxy_test.py
@@ -51,8 +51,6 @@
 )
 from flwr.server.compat.driver_client_proxy import DriverClientProxy, validate_task_res
 
-from .driver_client_proxy import DriverClientProxy
-
 MESSAGE_PARAMETERS = Parameters(tensors=[b"abc"], tensor_type="np")
 
 CLIENT_PROPERTIES = cast(Properties, {"tensor_type": "numpy.ndarray"})