From 60553bf2e9f65d40ae2ea8500a59c1949ef35946 Mon Sep 17 00:00:00 2001
From: Nitin Shivaraman <nitin.shivaraman@gmail.com>
Date: Fri, 17 Nov 2023 15:21:08 +0800
Subject: [PATCH] Fixed the comments of the PR

---
 src/qibotn/quimb.py              | 6 +++---
 tests/test_qasm_quimb_backend.py | 8 ++++----
 2 files changed, 7 insertions(+), 7 deletions(-)

diff --git a/src/qibotn/quimb.py b/src/qibotn/quimb.py
index 50c25ce..a9ac510 100644
--- a/src/qibotn/quimb.py
+++ b/src/qibotn/quimb.py
@@ -7,7 +7,7 @@ def from_qibo(circuit: QiboCircuit, is_mps: False, psi0=None, method='svd',
               cutoff=1e-6, cutoff_mode='abs'):
     nqubits = circuit.nqubits
     gate_opt = {}
-    if (is_mps):
+    if is_mps:
         tncirc = qtn.CircuitMPS(nqubits, psi0=psi0)
         gate_opt["method"] = method
         gate_opt["cutoff"] = cutoff
@@ -33,7 +33,7 @@ def init_state_tn(nqubits, init_state_sv):
     return qtn.tensor_1d.MatrixProductState.from_dense(init_state_sv, dims)
 
 
-def eval(qasm: str, init_state, backend="numpy"):
+def eval(qasm: str, init_state, is_mps, backend="numpy"):
     """Evaluate QASM with Quimb
 
     backend (quimb): numpy, cupy, jax. Passed to ``opt_einsum``.
@@ -41,7 +41,7 @@ def eval(qasm: str, init_state, backend="numpy"):
     """
     circuit = QiboCircuit.from_qasm(qasm)
     init_state_mps = init_state_tn(circuit.nqubits, init_state)
-    circ_quimb = from_qibo(circuit, is_mps=True, psi0=init_state_mps)
+    circ_quimb = from_qibo(circuit, is_mps, psi0=init_state_mps)
     interim = circ_quimb.psi.full_simplify(seq="DRC")
     amplitudes = interim.to_dense(backend=backend).flatten()
 
diff --git a/tests/test_qasm_quimb_backend.py b/tests/test_qasm_quimb_backend.py
index f46fbc6..f48d48a 100644
--- a/tests/test_qasm_quimb_backend.py
+++ b/tests/test_qasm_quimb_backend.py
@@ -30,9 +30,9 @@ def time(func):
     return time, res
 
 
-@pytest.mark.parametrize("nqubits, tolerance",
-                         [(1, 1e-6), (2, 1e-6), (5, 1e-3), (10, 1e-3)])
-def test_eval(nqubits: int, tolerance: float):
+@pytest.mark.parametrize("nqubits, tolerance, is_mps",
+                         [(1, 1e-6, True), (2, 1e-6, False), (5, 1e-3, True), (10, 1e-3, False)])
+def test_eval(nqubits: int, tolerance: float, is_mps: bool):
     # hack quimb to use the correct number of processes
     # TODO: remove completely, or at least delegate to the backend
     # implementation
@@ -55,7 +55,7 @@ def test_eval(nqubits: int, tolerance: float):
     # Test quimb
     quimb_time, result_tn = time(
         lambda: qibotn.quimb.eval(
-            qasm_circ, init_state_tn, backend=config.quimb.backend
+            qasm_circ, init_state_tn, is_mps, backend=config.quimb.backend
         )
     )