Reorder

adap · Aug 15, 2024 · ea3956a · ea3956a
1 parent cd3b888
commit ea3956a
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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(