misc: fix mpi import list

devitocodes · Nov 10, 2023 · 39a8486 · 39a8486
1 parent 730ed09
commit 39a8486
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/devito/__init__.py b/devito/__init__.py
@@ -22,7 +22,7 @@
 from devito.builtins import *  # noqa
 from devito.data.allocators import *  # noqa
 from devito.logger import error, warning, info, set_log_level  # noqa
-from devito.mpi import MPI  # noqa
+from devito.mpi import MPI, CustomTopology  # noqa
 try:
     from devito.checkpointing import DevitoCheckpoint, CheckpointOperator  # noqa
     from pyrevolve import Revolver

diff --git a/devito/mpi/distributed.py b/devito/mpi/distributed.py
@@ -17,8 +17,6 @@
 from devito.types.utils import DimensionTuple
 
 
-__all__ = ['CustomTopology']
-
 # Do not prematurely initialize MPI
 # This allows launching a Devito program from within another Python program
 # that has *already* initialized MPI
@@ -60,7 +58,7 @@ def __getattr__(self, name):
             return None
 
 
-__all__ = ['Distributor', 'SparseDistributor', 'MPI']
+__all__ = ['Distributor', 'SparseDistributor', 'MPI', 'CustomTopology']
 
 
 class AbstractDistributor(ABC):