Trying SimsoptRequires in place of monty.dev.requires for MpiPartition

hiddenSymmetries · Dec 23, 2021 · 7e2906d · 7e2906d
1 parent 572e3ba
commit 7e2906d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/simsopt/util/mpi.py b/src/simsopt/util/mpi.py
@@ -14,7 +14,7 @@
 from typing import Union
 import numpy as np
 
-from monty.dev import requires
+from .dev import SimsoptRequires
 try:
     from mpi4py import MPI
 except ImportError:
@@ -42,7 +42,7 @@ def log(level: int = logging.INFO):
 logger = logging.getLogger(__name__)
 
 
-@requires(MPI is not None, "mpi4py is not installed")
+@SimsoptRequires(MPI is not None, "mpi4py is not installed")
 class MpiPartition:
     """
     This module contains functions related to dividing up the set of