From 195fff3449c95a666d391a8a7ba40ed10e754adb Mon Sep 17 00:00:00 2001 From: Emma Ai Date: Thu, 22 Aug 2024 06:02:58 +0000 Subject: [PATCH] try single slice --- docker/requirements.txt | 2 +- odc/stats/plugins/lc_treelite_cultivated.py | 9 +++++---- 2 files changed, 6 insertions(+), 5 deletions(-) diff --git a/docker/requirements.txt b/docker/requirements.txt index 7da2769..7cff580 100644 --- 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 index 37825e0..73fa0ea 100644 --- 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)