Ensure dtype within _interp_points is floating

GlacioHack · Aug 16, 2024 · d36b713 · d36b713
1 parent c75ef7e
commit d36b713
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 0 deletions.
diff --git a/geoutils/raster/interpolate.py b/geoutils/raster/interpolate.py
@@ -247,6 +247,10 @@ def _interp_points(
 ) -> NDArrayNum | Callable[[tuple[NDArrayNum, NDArrayNum]], NDArrayNum]:
     """See description of Raster.interp_points."""
 
+    # If array is not a floating dtype (to support NaNs), convert dtype
+    if not np.issubdtype(array.dtype, np.floating):
+        array = array.astype(np.float32)
+
     # Get coordinates
     x, y = points
 

diff --git a/geoutils/raster/raster.py b/geoutils/raster/raster.py
@@ -3636,6 +3636,7 @@ def interp_points(
         :returns rpoints: Array of raster value(s) for the given points.
         """
 
+        # Extract array supporting NaNs
         array = self.get_nanarray()
         if self.count != 1:
             array = array[band - 1, :, :]