From 08ad49eb195b4fd13846665ad7867c32a99f520f Mon Sep 17 00:00:00 2001
From: MatteoRobbiati <robbiati.matteo@gmail.com>
Date: Mon, 3 Feb 2025 16:54:24 +0400
Subject: [PATCH] feat: make statevector calculation optional

---
 src/qibotn/backends/qmatchatea.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/qibotn/backends/qmatchatea.py b/src/qibotn/backends/qmatchatea.py
index ac3d934..74e29a3 100644
--- a/src/qibotn/backends/qmatchatea.py
+++ b/src/qibotn/backends/qmatchatea.py
@@ -77,6 +77,7 @@ class QMatchaTeaBackend(QibotnBackend):
         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 @@ class QMatchaTeaBackend(QibotnBackend):
             observables=observables,
         )
 
-        if circuit.num_qubits < 20:
+        if circuit.num_qubits < 20 and return_array:
             statevector = results.statevector
         else:
             statevector = None