From 9ae7dc18f53adfd3e0958020cbefc440934a1a6e Mon Sep 17 00:00:00 2001
From: tankya2 <tankya2@ihpc.a-star.edu.sg>
Date: Tue, 30 Jan 2024 17:49:30 +0800
Subject: [PATCH] Change execute_circuit inputs

---
 src/qibotn/backends/gpu.py | 17 ++++-------------
 1 file changed, 4 insertions(+), 13 deletions(-)

diff --git a/src/qibotn/backends/gpu.py b/src/qibotn/backends/gpu.py
index 47f667f..5f41344 100644
--- a/src/qibotn/backends/gpu.py
+++ b/src/qibotn/backends/gpu.py
@@ -1,14 +1,14 @@
 import numpy as np
 
 from qibo.backends.numpy import NumpyBackend
-from qibo.result import CircuitResult
+from qibo.states import CircuitResult
 from qibo.config import raise_error
 
 
 class CuTensorNet(NumpyBackend):  # pragma: no cover
     # CI does not test for GPU
 
-    def __init__(self, MPI_enabled=False, MPS_enabled=False, NCCL_enabled=False, expectation_enabled=False):
+    def __init__(self):
         super().__init__()
         import cuquantum  # pylint: disable=import-error
         from cuquantum import cutensornet as cutn  # pylint: disable=import-error
@@ -19,10 +19,6 @@ class CuTensorNet(NumpyBackend):  # pragma: no cover
         self.platform = "cutensornet"
         self.versions["cuquantum"] = self.cuquantum.__version__
         self.supports_multigpu = True
-        self.MPI_enabled = MPI_enabled
-        self.MPS_enabled = MPS_enabled
-        self.NCCL_enabled = NCCL_enabled
-        self.expectation_enabled = expectation_enabled
         self.handle = self.cutn.create()
 
     def apply_gate(self, gate, state, nqubits):  # pragma: no cover
@@ -57,7 +53,7 @@ class CuTensorNet(NumpyBackend):  # pragma: no cover
             raise TypeError("Type can be either complex64 or complex128")
 
     def execute_circuit(
-        self, circuit, initial_state=None, nshots=None, return_array=False
+        self, circuit, test, MPI_enabled=False, MPS_enabled=False, NCCL_enabled=False, expectation_enabled=False, initial_state=None, nshots=None, return_array=False
     ):  # pragma: no cover
         """Executes a quantum circuit.
 
@@ -72,12 +68,7 @@ class CuTensorNet(NumpyBackend):  # pragma: no cover
         """
 
         import qibotn.eval as eval
-
-        MPI_enabled = self.MPI_enabled
-        MPS_enabled = self.MPS_enabled
-        NCCL_enabled = self.NCCL_enabled
-        expectation_enabled = self.expectation_enabled
-
+        print("Test", test)
         if (
             MPI_enabled == False
             and MPS_enabled == False