Fixed a warning from numba.prange regarding implicit conversion fro…

…m `uint64` to `int64`. (#8)
sun-data · Nov 11, 2024 · 9224d1f · 9224d1f
1 parent 72c1b73
commit 9224d1f
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 0 deletions.
diff --git a/regridding/_conservative_ramshaw.py b/regridding/_conservative_ramshaw.py
@@ -185,6 +185,7 @@ def _sweep_axis(
         weight.append(w)
 
     for i in numba.prange(shape_sweep_x):
+        i = numba.types.int64(i)
 
         sweep_is_inside_static = False
         j = 0
@@ -846,6 +847,7 @@ def _grid_area_sweep(
     shape_output_x, shape_output_y = grid_x.shape
 
     for i in numba.prange(shape_output_x):
+        i = numba.types.int64(i)
 
         for j in range(shape_output_y - 1):
 

diff --git a/regridding/_regrid/_regrid_from_weights.py b/regridding/_regrid/_regrid_from_weights.py
@@ -120,6 +120,7 @@ def _regrid_from_weights(
 ) -> None:
 
     for d in numba.prange(len(weights)):
+        d = numba.types.int64(d)
         weights_d = weights[d]
         values_input_d = values_input[d].reshape(-1)
         values_output_d = values_output[d].reshape(-1)