diff --git a/src/qibotn/cutn.py b/src/qibotn/cutn.py index 4d5c3f2..b6213cf 100644 --- a/src/qibotn/cutn.py +++ b/src/qibotn/cutn.py @@ -291,7 +291,7 @@ def eval_tn_nccl_expectation(qibo_circ, datatype, n_samples=8): return result, rank -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 import socket from cuquantum import Network