Updated add_data tests passed qiskit-community#1268

Musa-Sina-Ertugrul · Nov 15, 2023 · e7f46c3 · e7f46c3
1 parent 2dbba8a
commit e7f46c3
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/qiskit_experiments/framework/experiment_data.py b/qiskit_experiments/framework/experiment_data.py
@@ -808,8 +808,8 @@ def add_data(
                                 else:
                                     experiment_seperator[inner_datum[0]["metadata"]["experiment_type"]] = ExperimentData()
                                     experiment_seperator[inner_datum[0]["metadata"]["experiment_type"]].add_data(inner_datum[0])
-                    
-                    self._result_data.append(datum)
+
+                        self._result_data.append(datum)
 
                 elif isinstance(datum, Result):
                     if datum["metadata"]:
@@ -821,6 +821,8 @@ def add_data(
             if composite_flag:
                 tmp_exp_data._set_child_data(list(experiment_seperator.values()))
                 self._set_child_data([tmp_exp_data])
+
+        return tmp_exp_data
 
     def __add_data(
         self,