more compact form for mps settings

src/qibotn/eval_qu.py

@@ -20,14 +20,11 @@ def dense_vector_tn_qu(qasm: str, initial_state, mps_opts, backend="numpy"):
         nqubits = int(np.log2(len(initial_state)))
         initial_state = init_state_tn(nqubits, initial_state)
 
-    if mps_opts:
+    circ_cls = qtn.circuit.CircuitMPS if mps_opts else qtn.circuit.Circuit
-        circ_quimb = qtn.circuit.CircuitMPS.from_openqasm2_str(
+    circ_quimb = circ_cls.from_openqasm2_str(
         qasm, psi0=initial_state, gate_opts=mps_opts
     )
 
-    else:
-        circ_quimb = qtn.circuit.Circuit.from_openqasm2_str(qasm, psi0=initial_state)
-
     interim = circ_quimb.psi.full_simplify(seq="DRC")
     amplitudes = interim.to_dense(backend=backend)
 

tests/test_quimb_backend.py

@@ -55,13 +55,11 @@ def test_eval(nqubits: int, tolerance: float, is_mps: bool):
         gate_opt["method"] = "svd"
         gate_opt["cutoff"] = 1e-6
         gate_opt["cutoff_mode"] = "abs"
+    else:
+        gate_opt = None
     result_tn = qibotn.eval_qu.dense_vector_tn_qu(
         qasm_circ, init_state_tn, gate_opt, backend=config.quimb.backend
     ).flatten()
-    else:
-        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