From b8f23d8b4b1e0a5d3531936e7105e127d616025e Mon Sep 17 00:00:00 2001 From: Santiago Martinez Balvanera Date: Tue, 4 Aug 2026 08:49:08 +0100 Subject: [PATCH] Avoid np stack error for empty arrays --- src/batdetect2/outputs/formats/raw.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/batdetect2/outputs/formats/raw.py b/src/batdetect2/outputs/formats/raw.py index c4150df..9a2e6d0 100644 --- a/src/batdetect2/outputs/formats/raw.py +++ b/src/batdetect2/outputs/formats/raw.py @@ -140,7 +140,7 @@ class RawFormatter(OutputFormatterProtocol[ClipDetections]): "clip_id": str(clip.uuid), } - if self.include_class_scores: + if self.include_class_scores and values["class_scores"]: class_scores = np.stack(values["class_scores"], axis=0) data_vars["class_scores"] = ( ["detection", "classes"], @@ -148,7 +148,7 @@ class RawFormatter(OutputFormatterProtocol[ClipDetections]): ) coords["classes"] = ("classes", self.targets.class_names) - if self.include_features: + if self.include_features and values["features"]: features = np.stack(values["features"], axis=0) data_vars["features"] = (["detection", "feature"], features) coords["feature"] = ("feature", np.arange(num_features))