Tests passed , Updated recursive approach, Updated add_data qiskit-co…

…mmunity#1268
Musa-Sina-Ertugrul · Apr 22, 2024 · fc3efea · fc3efea
1 parent f9ab8a5
commit fc3efea
Showing 1 changed file with 10 additions and 3 deletions.
diff --git a/qiskit_experiments/framework/experiment_data.py b/qiskit_experiments/framework/experiment_data.py
@@ -905,25 +905,32 @@ def add_data(
                             for inner_datum in datum["metadata"]["composite_metadata"]:
                                 if "composite_index" in inner_datum:
                                     for sub_expdata in composite_expdata:
-                                        sub_expdata.add_data(inner_datum)
+                                        self.add_data(inner_datum)
                         except IndexError or RuntimeError or AnalysisError:
                             new_child = ExperimentData()
                             for inner_datum in marginalized_datum:
                                 new_child.add_data(inner_datum)
+
+                        self._result_data.append(datum)
 
                     elif "composite_metadata" in datum:
+
                         marginalized_datum = self._marginalized_component_data([datum])
                         try:
                             composite_index = datum["composite_index"]
                             composite_expdata = [self.child_data(i) for i in composite_index]
                             for sub_expdata, sub_data in zip(composite_expdata, marginalized_datum):
                                 sub_expdata.add_data(sub_data)
+                            for inner_datum in datum["composite_metadata"]:
+                                if "composite_index" in inner_datum:
+                                    for sub_expdata in composite_expdata:
+                                        self.add_data(inner_datum)
                         except IndexError or RuntimeError or AnalysisError:
                             new_child = ExperimentData()
                             for inner_datum in marginalized_datum:
                                 new_child.add_data(inner_datum)
-
-                    self._result_data.append(datum)
+                    else:
+                        self._result_data.append(datum)
 
                 elif isinstance(datum, Result):
                     self._add_result_data(datum)