Update function name

qiboteam · Jan 24, 2024 · aba58c2 · aba58c2
1 parent c3586b4
commit aba58c2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/qibotn/cutn.py b/src/qibotn/cutn.py
@@ -12,7 +12,7 @@ def eval(qibo_circ, datatype):
     myconvertor = QiboCircuitToEinsum(qibo_circ, dtype=datatype)
     return contract(*myconvertor.state_vector_operands())
 
-def eval_tn_MPI_2(qibo_circ, datatype, n_samples=8):
+def eval_tn_MPI(qibo_circ, datatype, n_samples=8):
     from mpi4py import MPI 
     from cuquantum import Network
 
@@ -142,7 +142,7 @@ def eval_expectation(qibo_circ, datatype):
         *myconvertor.expectation_operands(PauliStringGen(qibo_circ.nqubits))
     )
 
-def eval_tn_MPI_2_expectation(qibo_circ, datatype, n_samples=8):
+def eval_tn_MPI_expectation(qibo_circ, datatype, n_samples=8):
     from mpi4py import MPI  # this line initializes MPI
     from cuquantum import Network