fix import error

gbrammer · Apr 29, 2024 · f36f77d · f36f77d
1 parent e06420f
commit f36f77d
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 3 deletions.
diff --git a/msaexp/resample_numba.py b/msaexp/resample_numba.py
@@ -6,6 +6,7 @@
     "resample_template_numba",
     "sample_gaussian_line_numba",
     "pixel_integrated_gaussian_numba",
+    "compute_igm"
 ]
 
 
@@ -204,7 +205,7 @@ def pixel_integrated_gaussian_numba(x, mu, sigma, dx=None, normalization=1.0):
 
 
 @jit(nopython=True, fastmath=True, error_model="numpy")
-def calculate_igm(z, wobs, scale_tau=1.):
+def compute_igm(z, wobs, scale_tau=1.):
     """
     Calculate Inoue+14 IGM transmission
     

diff --git a/msaexp/spectrum.py b/msaexp/spectrum.py
@@ -57,8 +57,8 @@ def float_representer(dumper, value):
 from . import utils as msautils
 
 try:
-    from .resample_numba import calculate_igm
-    igm_func = calculate_igm
+    from .resample_numba import compute_igm
+    igm_func = compute_igm
 
 except ImportError:
     igm = eazy.igm.Inoue14()