Minor fixes in imports

hiddenSymmetries · Jan 29, 2023 · 1ffef6e · 1ffef6e
1 parent 04e330c
commit 1ffef6e
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/simsopt/geo/boozersurface.py b/src/simsopt/geo/boozersurface.py
@@ -2,7 +2,7 @@
 from scipy.linalg import lu
 from scipy.optimize import minimize, least_squares
 
-from ..geo.surfaceobjectives import boozer_surface_residual
+from .surfaceobjectives import boozer_surface_residual
 from .._core.optimizable import Optimizable
 
 __all__ = ['BoozerSurface']

diff --git a/src/simsopt/geo/surfaceobjectives.py b/src/simsopt/geo/surfaceobjectives.py
@@ -6,8 +6,8 @@
 import simsoptpp as sopp
 from .._core.optimizable import Optimizable
 from .._core.derivative import Derivative, derivative_dec
-from ..geo.surface import Surface
-from ..geo.surfacexyztensorfourier import SurfaceXYZTensorFourier
+from .surface import Surface
+from .surfacexyztensorfourier import SurfaceXYZTensorFourier
 from ..objectives.utilities import forward_backward
 
 __all__ = ['Area', 'Volume', 'ToroidalFlux', 'PrincipalCurvature',