Fixed issue in simulation.py

jfuruness · Dec 13, 2024 · d3ace43 · d3ace43
1 parent 9318409
commit d3ace43
Showing 1 changed file with 10 additions and 2 deletions.
diff --git a/bgpy/simulation_framework/simulation.py b/bgpy/simulation_framework/simulation.py
@@ -287,12 +287,20 @@ def _get_data(self):
         if self.parse_cpus == 1:
             # Results are a list of lists of metric trackers that we then sum
             return sum(
-                self._get_single_process_results(), start=self.GraphDataAggregatorCls()
+                self._get_single_process_results(),
+                start=self.GraphDataAggregatorCls(
+                    graph_categories=self.graph_categories
+                ),
             )
         # Multiprocess
         else:
             # Results are a list of lists of metric trackers that we then sum
-            return sum(self._get_mp_results(), start=self.GraphDataAggregatorCls())
+            return sum(
+                self._get_mp_results(),
+                start=self.GraphDataAggregatorCls(
+                    graph_categories=self.graph_categories
+                ),
+            )
 
     ###########################
     # Multiprocessing Methods #