From a9408dd4e211f82e8b0e61e32bcfc7c386086285 Mon Sep 17 00:00:00 2001 From: Heeji Han Date: Mon, 10 Aug 2026 06:03:45 +0000 Subject: [PATCH] [fix] fix image pipeline handling of missing decoded feature tensor hash --- .../pipelines/split_inference/image_split_inference.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/compressai_vision/pipelines/split_inference/image_split_inference.py b/compressai_vision/pipelines/split_inference/image_split_inference.py index 074638c..32fc25b 100644 --- a/compressai_vision/pipelines/split_inference/image_split_inference.py +++ b/compressai_vision/pipelines/split_inference/image_split_inference.py @@ -286,7 +286,10 @@ def __call__( out_res["coded_order"] = e out_res["org_input_size"] = f"{d[0]['height']}x{d[0]['width']}" out_res["input_size"] = dec_features["input_size"][0] - if "decoded_feature_tensor_hash" in dec_features: + if ( + "decoded_feature_tensor_hash" in dec_features + and dec_features["decoded_feature_tensor_hash"] is not None + ): out_res["decoded_feature_tensor_hash"] = ( dec_features["decoded_feature_tensor_hash"][e] if e in dec_features["decoded_feature_tensor_hash"]