added function name to __all__

hiddenSymmetries · Mar 29, 2024 · 5663c88 · 5663c88
1 parent 284c4ea
commit 5663c88
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/src/simsopt/field/__init__.py b/src/simsopt/field/__init__.py
@@ -17,4 +17,5 @@
     + mgrid.__all__
     + normal_field.__all__
     + tracing.__all__
+    + magnetic_axis_helpers.__all__
 )
diff --git a/src/simsopt/field/magnetic_axis_helpers.py b/src/simsopt/field/magnetic_axis_helpers.py
@@ -2,6 +2,8 @@
 from simsopt.geo import CurveRZFourier
 from scipy.integrate import solve_ivp
 
+__all__ = ['compute_on_axis_iota']
+
 def compute_on_axis_iota(axis, magnetic_field):
     """
     Computes the rotational transform on the magnetic axis of a device using a method based on