From 9a0ea93d457bd7f4e141725786e0ae7097ea502a Mon Sep 17 00:00:00 2001
From: NimaSarajpoor <nimasarajpoor@gmail.com>
Date: Tue, 2 Jul 2024 08:12:43 -0400
Subject: [PATCH] Fix missing decorator

---
 stumpy/maamp.py  | 1 +
 stumpy/mstump.py | 1 +
 2 files changed, 2 insertions(+)

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,