use min_scalar_type in rasterize (#97)

xarray-contrib · Feb 6, 2025 · f035e23 · f035e23
1 parent 131e6cb
commit f035e23
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/xvec/zonal.py b/xvec/zonal.py
@@ -52,7 +52,7 @@ def _zonal_stats_rasterize(
 
     transform = acc._obj.rio.transform()
     length = len(geometry)
-    dtype = np.int16 if length < np.iinfo(np.int16).max else np.int32
+    dtype = np.min_scalar_type(length + 1)
 
     labels = features.rasterize(
         zip(geometry, range(length), strict=False),