check if masked-array

ioos · Mar 20, 2024 · dfa9d08 · dfa9d08
1 parent 5611b64
commit dfa9d08
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/ioos_qc/qartod.py b/ioos_qc/qartod.py
@@ -738,8 +738,9 @@ 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):
+            except (ImportError, TypeError, numba.core.errors.NumbaTypeError):
                 return w.apply(np.ptp, raw=True)
         check_func = np.ptp
     else: