try single slice

opendatacube · Aug 22, 2024 · 195fff3 · 195fff3
1 parent 083c1c8
commit 195fff3
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 5 deletions.
diff --git a/docker/requirements.txt b/docker/requirements.txt
@@ -11,7 +11,7 @@ odc-dscache>=0.2.3
 odc-stac @ git+https://github.com/opendatacube/odc-stac@69bdf64
 
 # odc-stac is in PyPI
-odc-stats[ows] @ git+https://github.com/opendatacube/odc-stats@b404842
+odc-stats[ows] @ git+https://github.com/opendatacube/odc-stats@083c1c8
 
 # For ML
 tflite-runtime

diff --git a/odc/stats/plugins/lc_treelite_cultivated.py b/odc/stats/plugins/lc_treelite_cultivated.py
@@ -246,10 +246,11 @@ def aggregate_results(input_block, cultivated_value, natural_value):
     m_size = input_block.shape[-1]
 
     if m_size > 1:
-        output_block = ne.evaluate(
-            "sum(m,axis=2)",
-            local_dict={"m": output_block},
-        ).astype("float32")
+        output_block = input_block[..., -1]
+        # output_block = ne.evaluate(
+        #     "sum(m,axis=2)",
+        #     local_dict={"m": output_block},
+        # ).astype("float32")
 
     else:
         output_block = output_block.squeeze(axis=-1)