diff --git a/src/py/flwr/client/process/process.py b/src/py/flwr/client/process/process.py index 5c7de7c0015e..2e1cb50818e6 100644 --- a/src/py/flwr/client/process/process.py +++ b/src/py/flwr/client/process/process.py @@ -72,7 +72,7 @@ def on_channel_state_change(channel_connectivity: str) -> None: stub = ClientAppIoStub(channel) # Pull Message, Context, and Run from SuperNode - run, message, context = pull_message(stub=stub, token=token) + message, context, run = pull_message(stub=stub, token=token) load_client_app_fn = _get_load_client_app_fn( default_app_ref="", @@ -97,15 +97,15 @@ 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[Message, Context, Run]: """Pull message from SuperNode to ClientApp.""" res: PullClientAppInputsResponse = stub.PullClientAppInputs( PullClientAppInputsRequest(token=token) ) - run = run_from_proto(res.run) message = message_from_proto(res.message) context = context_from_proto(res.context) - return run, message, context + run = run_from_proto(res.run) + return message, context, run def push_message(