refactoring, added tests, fixed minor bug

johannesulf · Nov 25, 2024 · c391f95 · c391f95
1 parent 2f5171d
commit c391f95
Show file tree

Hide file tree

Showing 7 changed files with 458 additions and 406 deletions.
diff --git a/nautilus/bounds/__init__.py b/nautilus/bounds/__init__.py
@@ -1,8 +1,10 @@
 """Modules implementing various multi-dimensional bounds."""
 
 from .basic import UnitCube, Ellipsoid, UnitCubeEllipsoidMixture
-from .neural import NeuralBound, NautilusBound
+from .nautilus import NautilusBound
+from .neural import NeuralBound
+from .periodic import PhaseShift
 from .union import Union
 
-__all__ = ['UnitCube', 'Ellipsoid', 'UnitCubeEllipsoidMixture', 'Union',
-           'NeuralBound', 'NautilusBound']
+__all__ = ['Ellipsoid', 'NautilusBound', 'NeuralBound',
+           'PhaseShift', 'Union', 'UnitCube', 'UnitCubeEllipsoidMixture']