Use input round_number (#1292)

Signed-off-by: Lerer, Eran <[email protected]>
securefederatedai · Jan 21, 2025 · f18b685 · f18b685
1 parent a19f869
commit f18b685
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",