Assert len of outputs for repeats.

SchmidtDSE · Dec 5, 2024 · 3158a5a · 3158a5a
1 parent 1228b00
commit 3158a5a
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/sim_tasks.py b/sim_tasks.py
@@ -1371,8 +1371,13 @@ def _process_outputs(self, output_sets_realized):
         Args:
             output_sets_realized: Sets of the individual simulation outputs.
         """
+        assert len(output_sets_realized) == self.get_iterations()
+
         def simplify_and_combine(outputs_realized):
-            simplified = map(lambda x: self._simplify_record(x), outputs_realized)
+            simplified = map(
+                lambda x: self._simplify_record(x),
+                itertools.chain(*outputs_realized)
+            )
             return functools.reduce(lambda a, b: self._combine(a, b), simplified)
 
         def get_stats(key, summaries):