Use njit to silence new numba warnings

moble · Jun 4, 2024 · 971ca78 · 971ca78
1 parent a3b928e
commit 971ca78
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/scri/utilities.py b/scri/utilities.py
@@ -372,7 +372,7 @@ def shuffle(a):
                     b_array_bit += shuffle_width
             return b
 
-        return nb.jit(shuffle)
+        return nb.njit(shuffle)
 
     else:
         # This function is almost the same as above, except for:
@@ -404,4 +404,4 @@ def unshuffle(b):
                     b_array_bit += shuffle_width
             return a
 
-        return nb.jit(unshuffle)
+        return nb.njit(unshuffle)