Fixed chunk size

Author: seanlaw

stumpy/scraamp.py

@@ -434,8 +434,9 @@ def __init__(
             self._diags, self._m, self._n_A, self._n_B
         )
         self._chunk_diags_ranges = core._get_array_ranges(
-            self._ndist_counts, self._n_chunks
+            self._ndist_counts, self._n_chunks, truncate=True
         )
+        self._n_chunks = self._chunk_diags_ranges.shape[0]
         self._chunk_idx = 0
 
     def update(self):

stumpy/scrump.py

@@ -451,8 +451,9 @@ def __init__(
             self._diags, self._m, self._n_A, self._n_B
         )
         self._chunk_diags_ranges = core._get_array_ranges(
-            self._ndist_counts, self._n_chunks
+            self._ndist_counts, self._n_chunks, truncate=True
         )
+        self._n_chunks = self._chunk_diags_ranges.shape[0]
         self._chunk_idx = 0
 
     def update(self):

test.sh

@@ -54,6 +54,8 @@ check_print()
 
 test_custom()
 {
+    # export NUMBA_DISABLE_JIT=1
+    # export NUMBA_ENABLE_CUDASIM=1
     # Test one or more user-defined functions repeatedly
     for VARIABLE in {1..10}
     do