From 151a5a51a1c653b33a93940a1acaf84e7418895a Mon Sep 17 00:00:00 2001
From: vinitha-balachandran
 <balachan@x1000c1s1b0n1.hostmgmt2000.cm.asp2a.nscc.sg>
Date: Thu, 22 Feb 2024 18:11:48 +0800
Subject: [PATCH] Change output to QuantumState class for quimb backend

---
 src/qibotn/backends/quimb.py | 5 ++---
 1 file changed, 2 insertions(+), 3 deletions(-)

diff --git a/src/qibotn/backends/quimb.py b/src/qibotn/backends/quimb.py
index cca6799..31e24c7 100644
--- a/src/qibotn/backends/quimb.py
+++ b/src/qibotn/backends/quimb.py
@@ -1,6 +1,6 @@
 from qibo.backends.numpy import NumpyBackend
 from qibo.config import raise_error
-from qibo.result import CircuitResult
+from qibo.result import QuantumState
 
 
 class QuimbBackend(NumpyBackend):
@@ -80,5 +80,4 @@ class QuimbBackend(NumpyBackend):
         if return_array:
             return state.flatten()
         else:
-            circuit._final_state = CircuitResult(self, circuit, state.flatten(), nshots)
-            return circuit._final_state
+            return QuantumState(state.flatten())