From 5663c88cfd68e440a52e08d50f951495819221bc Mon Sep 17 00:00:00 2001
From: Andrew Giuliani <giuliani@cims.nyu.edu>
Date: Fri, 29 Mar 2024 11:10:06 -0400
Subject: [PATCH] added function name to __all__

---
 src/simsopt/field/__init__.py              | 1 +
 src/simsopt/field/magnetic_axis_helpers.py | 2 ++
 2 files changed, 3 insertions(+)

diff --git a/src/simsopt/field/__init__.py b/src/simsopt/field/__init__.py
index d4f85d894..8e652395e 100644
--- 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
index ee3ef1809..67ac71129 100644
--- 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