added flatten of state in pytest instead of using within eval_qu

2024-02-08 16:22:38 +08:00
parent 764c0bd857
commit 773953e18b
1 changed files with 1 additions and 1 deletions
--- 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,