Fix return type

adap · Aug 15, 2024 · 3ae8021 · 3ae8021
1 parent 81c2a83
commit 3ae8021
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/py/flwr/client/process/process.py b/src/py/flwr/client/process/process.py
@@ -15,6 +15,7 @@
 """Flower background ClientApp."""
 
 from logging import DEBUG, ERROR, INFO
+from typing import Tuple
 
 import grpc
 
@@ -31,7 +32,7 @@
 )
 from flwr.common.typing import Run
 
-# pylint: disable=E0401,E0611
+# pylint: disable=E0611
 from flwr.proto.clientappio_pb2 import (
     PullClientAppInputsRequest,
     PullClientAppInputsResponse,
@@ -96,7 +97,7 @@ def on_channel_state_change(channel_connectivity: str) -> None:
         channel.close()
 
 
-def pull_message(stub: grpc.Channel, token: int) -> tuple[Run, Message, Context]:
+def pull_message(stub: grpc.Channel, token: int) -> Tuple[Run, Message, Context]:
     """Pull message from SuperNode to ClientApp."""
     res: PullClientAppInputsResponse = stub.PullClientAppInputs(
         PullClientAppInputsRequest(token=token)