Merge pull request #114 from wri/fix-distalert-no-context

Flat dist alert structure for cases without context layer

tests/test_dist_alerts.py

@@ -143,3 +143,20 @@ def test_dist_alert_tool_buffer():
     assert (
         result["natural short vegetation"] < result_buffered["natural short vegetation"]
     )
+
+
+def test_dist_alert_tool_no_context():
+    result = tools.dist_alerts_tool.invoke(
+        input={
+            "name": "BRA.13.369_2",
+            "gadm_id": "BRA.13.369_2",
+            "gadm_level": 2,
+            "context_layer_name": None,
+            "threshold": 8,
+            "min_date": datetime.date(2021, 8, 12),
+            "max_date": datetime.date(2024, 8, 12),
+        }
+    )
+
+    assert len(result) == 1
+    assert "disturbances" in result

zeno/agents/distalert/tools.py

@@ -173,9 +173,7 @@ def get_distalerts_unfiltered(
         scale=DIST_ALERT_STATS_SCALE,
     ).getInfo()
 
-    zone_stats_result = {"context_layer_name": None}
-    for feat in zone_stats["features"]:
-        zone_stats_result[name] = {"disturbances": feat["properties"]["sum"]}
+    zone_stats_result= {"disturbances": zone_stats["features"][0]["properties"]["sum"]}
 
     vectorize = (
         distalerts.gte(threshold).updateMask(distalerts.gte(threshold)).selfMask()
@@ -233,6 +231,8 @@ def dist_alerts_tool(
 
     This tool quantifies vegetation disturbance alerts over an area of interest
     and summarizes the alerts in statistics by context layer types.
+
+    The unit of disturbances that are returned are hectares.
     """
     print("---DIST ALERTS TOOL---")