Init next tensors dict (#1293)

cloudnoize · web-flow · commit 810414436f26 · 2025-01-21T14:00:33.000+01:00
Signed-off-by: Lerer, Eran &lt;eran.lerer@intel.com&gt;
diff --git a/openfl/component/aggregator/aggregator.py b/openfl/component/aggregator/aggregator.py
@@ -352,6 +352,7 @@ def _save_model(self, round_number, file_path):
         ]
         tensor_dict = {}
         tensor_tuple_dict = {}
+        next_round_tensors = {}
         for tk in tensor_keys:
             tk_name, _, _, _, _ = tk
             tensor_value = self.tensor_db.get_tensor_from_cache(tk)

Original file line number	Diff line number	Diff line change
`@@ -352,6 +352,7 @@ def _save_model(self, round_number, file_path):`
`352`	`352`	`]`
`353`	`353`	`tensor_dict = {}`
`354`	`354`	`tensor_tuple_dict = {}`
	`355`	`+ next_round_tensors = {}`
`355`	`356`	`for tk in tensor_keys:`
`356`	`357`	`tk_name, _, _, _, _ = tk`
`357`	`358`	`tensor_value = self.tensor_db.get_tensor_from_cache(tk)`