From 52da762e1b3f328b1da5f6e22c7a10a5d5c5ae82 Mon Sep 17 00:00:00 2001
From: tankya2 <tankya2@ihpc.a-star.edu.sg>
Date: Thu, 22 Feb 2024 17:19:39 +0800
Subject: [PATCH] Change output to QuantumState class

---
 src/qibotn/backends/cutensornet.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/src/qibotn/backends/cutensornet.py b/src/qibotn/backends/cutensornet.py
index 104b41c..b776dec 100644
--- a/src/qibotn/backends/cutensornet.py
+++ b/src/qibotn/backends/cutensornet.py
@@ -2,7 +2,7 @@ import cuquantum  # pylint: disable=import-error
 import numpy as np
 from qibo.backends.numpy import NumpyBackend
 from qibo.config import raise_error
-from qibo.result import CircuitResult
+from qibo.result import QuantumState
 
 CUDA_TYPES = {
     "complex64": (
@@ -185,5 +185,5 @@ class CuTensorNet(NumpyBackend):  # pragma: no cover
         if return_array:
             return state.flatten()
         else:
-            circuit._final_state = CircuitResult(self, circuit, state.flatten(), nshots)
-            return circuit._final_state
+            return QuantumState(state.flatten())
+