reverting after fixes

adap · Feb 29, 2024 · 62c1dce · 62c1dce
1 parent 708d4ce
commit 62c1dce
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 4 deletions.
diff --git a/src/py/flwr/client/message_handler/message_handler.py b/src/py/flwr/client/message_handler/message_handler.py
@@ -86,7 +86,7 @@ def handle_control_message(message: Message) -> Tuple[Optional[Message], int]:
         reason = cast(int, disconnect_msg.disconnect_res.reason)
         recordset = RecordSet()
         recordset.configs_records["config"] = ConfigsRecord({"reason": reason})
-        out_message = message.create_reply(content=recordset, ttl="")
+        out_message = message.create_reply(recordset, ttl="")
         # Return TaskRes and sleep duration
         return out_message, sleep_duration
 
@@ -148,7 +148,7 @@ def handle_legacy_message_from_msgtype(
         raise ValueError(f"Invalid message type: {message_type}")
 
     # Return Message
-    return message.create_reply(content=out_recordset, ttl="")
+    return message.create_reply(out_recordset, ttl="")
 
 
 def _reconnect(

diff --git a/src/py/flwr/client/mod/secure_aggregation/secaggplus_mod.py b/src/py/flwr/client/mod/secure_aggregation/secaggplus_mod.py
@@ -189,7 +189,7 @@ def secaggplus_mod(
 
     # Return message
     content = RecordSet(configs_records={RECORD_KEY_CONFIGS: ConfigsRecord(res, False)})
-    return msg.create_reply(content=content, ttl="")
+    return msg.create_reply(content, ttl="")
 
 
 def check_stage(current_stage: str, configs: ConfigsRecord) -> None:

diff --git a/src/py/flwr/client/mod/secure_aggregation/secaggplus_mod_test.py b/src/py/flwr/client/mod/secure_aggregation/secaggplus_mod_test.py
@@ -38,7 +38,7 @@ def get_test_handler(
     """."""
 
     def empty_ffn(_msg: Message, _2: Context) -> Message:
-        return _msg.create_reply(content=RecordSet(), ttl="")
+        return _msg.create_reply(RecordSet(), ttl="")
 
     app = make_ffn(empty_ffn, [secaggplus_mod])