From 773953e18b6c3b3ecea85f6ad3dca2e37c04d02e 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 e01cde47..81a0e2b7 100644
--- a/tests/test_quimb_backend.py
+++ b/tests/test_quimb_backend.py
@@ -53,7 +53,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,