From 91176eb1cc0ad21843e31a678332dabb0c79740f Mon Sep 17 00:00:00 2001
From: Ryan Grout <ryan@ryangrout.org>
Date: Mon, 5 Feb 2024 13:23:00 -0700
Subject: [PATCH] Cache jitted function.

---
 pysheds/_sgrid.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pysheds/_sgrid.py b/pysheds/_sgrid.py
index 4dc91d2..e0db269 100644
--- a/pysheds/_sgrid.py
+++ b/pysheds/_sgrid.py
@@ -1919,7 +1919,7 @@ def count(start=0, step=1):
         yield n
         n += step
 
-@njit
+@njit(cache=True)
 def _priority_flood(dem, dem_mask, tuple_type):
     open_cells = typedlist.List.empty_list(tuple_type)  # Priority queue
     pits = typedlist.List.empty_list(tuple_type)  # FIFO queue