Return function output directly

2023-08-03 10:27:19 +08:00
parent 73d65a6af1
commit fc3e0c24bd
1 changed files with 4 additions and 3 deletions
--- a/src/qibotn/cutn.py
+++ b/src/qibotn/cutn.py
@@ -13,6 +13,7 @@ def eval(qibo_circ, datatype):
 def eval_mps(qibo_circ, gate_algo, datatype):
    myconvertor = QiboCircuitToMPS(qibo_circ, gate_algo, dtype=datatype)
    mps_helper = MPSContractionHelper(myconvertor.num_qubits)
-    sv_mps = mps_helper.contract_state_vector(
-        myconvertor.mps_tensors, myconvertor.options)
-    return sv_mps
+
+    return mps_helper.contract_state_vector(
+        myconvertor.mps_tensors, myconvertor.options
+    )