pass node to node field

adap · Sep 27, 2024 · da0156f · da0156f
1 parent fa3c285
commit da0156f
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/py/flwr/client/connection/grpc_rere/grpc_rere_connection.py b/src/py/flwr/client/connection/grpc_rere/grpc_rere_connection.py
@@ -236,7 +236,7 @@ def send(self, message: Message) -> None:
         task_res = message_to_taskres(message)
 
         # Serialize ProtoBuf to bytes
-        req = PushTaskResRequest(task_res_list=[task_res])
+        req = PushTaskResRequest(node=self.node, task_res_list=[task_res])
         self.retrier.invoke(self.api.PushTaskRes, req)
 
         # Cleanup
@@ -245,7 +245,7 @@ def send(self, message: Message) -> None:
     def get_run(self, run_id: int) -> Run:
         """Get run info."""
         # Call FleetAPI
-        req = GetRunRequest(run_id=run_id)
+        req = GetRunRequest(node=self.node, run_id=run_id)
         res: GetRunResponse = self.retrier.invoke(
             self.api.GetRun,
             request=req,
@@ -263,7 +263,7 @@ def get_run(self, run_id: int) -> Run:
     def get_fab(self, fab_hash: str) -> Fab:
         """Get FAB file."""
         # Call FleetAPI
-        req = GetFabRequest(hash_str=fab_hash)
+        req = GetFabRequest(node=self.node, hash_str=fab_hash)
         res: GetFabResponse = self.retrier.invoke(
             self.api.GetFab,
             request=req,