Fix vectorization by group

wri · Jan 27, 2025 · b6d9920 · b6d9920
1 parent 5c2dbf3
commit b6d9920
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tests/test_dist_alerts.py b/tests/test_dist_alerts.py
@@ -11,8 +11,8 @@ def test_dist_alert_tool():
             "gadm_level": 2,
             "context_layer_name": "WRI/SBTN/naturalLands/v1/2020",
             "threshold": 8,
-            "min_date": datetime.date(2021, 8, 12),
-            "max_date": datetime.date(2024, 8, 12),
+            "min_date": datetime.date(2023, 8, 1),
+            "max_date": datetime.date(2023, 8, 31),
         }
     )
 

diff --git a/zeno/agents/distalert/tool_dist_alerts.py b/zeno/agents/distalert/tool_dist_alerts.py
@@ -113,7 +113,7 @@ def get_zone_stats(
         collection=gee_features,
         reducer=ee.Reducer.count().group(groupField=1, groupName=choice["band"]),
         scale=choice["resolution"],
-    ).getInfo(), zone_stats_img
+    ).getInfo(), zone_stats_img.select(choice["band"])
 
 
 def get_alerts_by_context_layer(