handle non-numba case

ioos · Mar 20, 2024 · b6e8a84 · b6e8a84
1 parent 6a3e5ec
commit b6e8a84
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ioos_qc/qartod.py b/ioos_qc/qartod.py
@@ -10,6 +10,11 @@
 import numpy as np
 import pandas as pd
 
+try:
+    from numba.core.errors import NumbaTypeError
+except ImportError:
+    NumbaTypeError = TypeError
+
 from ioos_qc.utils import (
     isnan,
     isfixedlength,
@@ -738,9 +743,8 @@ def attenuated_signal_test(inp : Sequence[N],
         def window_func(w):
             # When pandas>=1.0 and numba are installed, this is about twice as fast
             try:
-                import numba
                 return w.apply(np.ptp, raw=True, engine='numba')
-            except (ImportError, TypeError, numba.core.errors.NumbaTypeError):
+            except (ImportError, TypeError, NumbaTypeError):
                 return w.apply(np.ptp, raw=True)
         check_func = np.ptp
     else: