From 3b73a22e6a1d3069ce9bec899df3ccd33bc18ac1 Mon Sep 17 00:00:00 2001
From: vinitha-balachandran <vinitha-balachandran@ihpc.a-star.edu.sg>
Date: Thu, 8 Feb 2024 16:22:38 +0800
Subject: [PATCH] added flatten of state in pytest instead of using within
 eval_qu

---
 tests/test_quimb_backend.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_quimb_backend.py b/tests/test_quimb_backend.py
index da4d422..15ba652 100644
--- a/tests/test_quimb_backend.py
+++ b/tests/test_quimb_backend.py
@@ -52,7 +52,7 @@ def test_eval(nqubits: int, tolerance: float, is_mps: bool):
     # Test quimb
     result_tn = qibotn.eval_qu.dense_vector_tn_qu(
         qasm_circ, init_state_tn, is_mps, backend=config.quimb.backend
-    )
+    ).flatten()
 
     assert np.allclose(
         result_sv, result_tn, atol=tolerance