feat: make statevector calculation optional

qiboteam · Feb 3, 2025 · 08ad49e · 08ad49e
1 parent 67b900b
commit 08ad49e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/qibotn/backends/qmatchatea.py b/src/qibotn/backends/qmatchatea.py
@@ -77,6 +77,7 @@ def execute_circuit(
         initial_state=None,
         nshots=None,
         prob_type="U",
+        return_array=False,
         **prob_kwargs,
     ):
         """Execute a Qibo quantum circuit using tensor network simulation.
@@ -153,7 +154,7 @@ def execute_circuit(
             observables=observables,
         )
 
-        if circuit.num_qubits < 20:
+        if circuit.num_qubits < 20 and return_array:
             statevector = results.statevector
         else:
             statevector = None