Add missing count stat

UM-RMRS · Oct 18, 2023 · b9aff66 · b9aff66
1 parent 3cbaf65
commit b9aff66
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 2 deletions.
diff --git a/raster_tools/zonal.py b/raster_tools/zonal.py
@@ -133,7 +133,18 @@ def _asm_entropy_fin(s, func):
 
 
 _DASK_STAT_NAMES = frozenset(
-    ("max", "mean", "median", "min", "prod", "size", "std", "sum", "var")
+    (
+        "count",
+        "max",
+        "mean",
+        "median",
+        "min",
+        "prod",
+        "size",
+        "std",
+        "sum",
+        "var",
+    )
 )
 _CUSTOM_STAT_NAMES = frozenset(("asm", "entropy", "mode", "nunique"))
 _CUSTOM_STAT_AGGS = {

diff --git a/tests/test_zonal.py b/tests/test_zonal.py
@@ -113,7 +113,7 @@ def rasters_to_zonal_df(feat_raster, data_raster):
     "stats",
     [
         ["max", "mean", "median", "min", "size", "std", "sum", "var"],
-        ["asm", "entropy", "mode", "nunique", "mean", "var"],
+        ["asm", "entropy", "mode", "nunique", "mean", "var", "count"],
         ["asm", "median"],
         "median",
         ["mean", "median"],