diff --git a/stumpy/maamp.py b/stumpy/maamp.py index 0bc9f8904..2de84cdcc 100644 --- a/stumpy/maamp.py +++ b/stumpy/maamp.py @@ -713,6 +713,7 @@ def _compute_multi_p_norm( core._apply_exclusion_zone(p_norm, idx, excl_zone, np.inf) +@njit def _maamp( T, m, diff --git a/stumpy/mstump.py b/stumpy/mstump.py index bdb1e8ada..c4d7fe0ab 100644 --- a/stumpy/mstump.py +++ b/stumpy/mstump.py @@ -950,6 +950,7 @@ def _compute_multi_D( core._apply_exclusion_zone(D, idx, excl_zone, np.inf) +@njit def _mstump( T, m,