diff --git a/src/py/flwr/client/message_handler/message_handler.py b/src/py/flwr/client/message_handler/message_handler.py index 66c4e7668776..87cace88ec27 100644 --- 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 index 521398d20145..63bf6bd6ebe2 100644 --- 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 index d0315be02fa8..a46a2fa1b16e 100644 --- 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])