From f18b6854e1a020282976f118a4be5677a239b0fd Mon Sep 17 00:00:00 2001 From: Eran Lerer Date: Tue, 21 Jan 2025 13:40:37 +0200 Subject: [PATCH] Use input round_number (#1292) Signed-off-by: Lerer, Eran --- openfl/component/aggregator/aggregator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py index c6829e75b9..d0fb33408c 100644 --- a/openfl/component/aggregator/aggregator.py +++ b/openfl/component/aggregator/aggregator.py @@ -373,7 +373,7 @@ def _save_model(self, round_number, file_path): self.next_model_round_number, ("model",) ) self.persistent_db.finalize_round( - tensor_tuple_dict, next_round_tensors, self.round_number, self.best_model_score + tensor_tuple_dict, next_round_tensors, round_number, self.best_model_score ) logger.info( "Persist model and clean task result for round %s",