Minor interface cleanup

DedalusProject · Jan 28, 2024 · ef0d652 · ef0d652
1 parent 7a7a998
commit ef0d652
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 21 deletions.
diff --git a/dedalus/core/__init__.py b/dedalus/core/__init__.py
@@ -1,4 +0,0 @@
-
-from . import distributor
-from . import domain
-#from . import spaces

diff --git a/dedalus/core/timesteppers.py b/dedalus/core/timesteppers.py
@@ -7,9 +7,14 @@
 from .system import CoeffSystem
 from ..tools.array import apply_sparse
 
+
+# Public interface
+__all__ = []
+
 # Track implemented schemes
 schemes = OrderedDict()
 def add_scheme(scheme):
+    __all__.append(scheme.__name__)
     schemes[scheme.__name__] = scheme
     return scheme
 

diff --git a/dedalus/dev.py b/dedalus/dev.py
diff --git a/dedalus/public.py b/dedalus/public.py
@@ -1,12 +1,12 @@
 """Dedalus public interface."""
 
 # Import public interfaces from submodules
+from .core.arithmetic import *
+from .core.basis import *
 from .core.coords import *
 from .core.distributor import *
-from .core.basis import *
 from .core.field import *
 from .core.operators import *
-from .core.arithmetic import *
 from .core.problems import *
 from .core.timesteppers import *