moved list

Signed-off-by: Melvin Strobl <[email protected]>
cirKITers · Nov 15, 2024 · 8bad289 · 8bad289
1 parent e8aa652
commit 8bad289
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/qml_essentials/model.py b/qml_essentials/model.py
@@ -595,9 +595,10 @@ def _forward(
                         inputs=inputs,
                     )
 
+        if isinstance(result, list):
+            result = np.stack(result)
+
         if self.execution_type == "expval" and self.output_qubit == -1:
-            if isinstance(result, list):
-                result = np.stack(result)
 
             # Calculating mean value after stacking, to not
             # discard gradient information