From 68351d2224999fc7272e1f1aafb776c8dc286266 Mon Sep 17 00:00:00 2001 From: ccarrizosa Date: Sat, 9 Nov 2024 22:35:11 +0100 Subject: [PATCH] Fix numpy exception handling --- batdetect2/utils/detector_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/batdetect2/utils/detector_utils.py b/batdetect2/utils/detector_utils.py index 8d6ca7f..d8942e0 100644 --- a/batdetect2/utils/detector_utils.py +++ b/batdetect2/utils/detector_utils.py @@ -242,7 +242,7 @@ def format_single_result( ) class_name = class_names[np.argmax(class_overall)] annotations = get_annotations_from_preds(predictions, class_names) - except (np.AxisError, ValueError): + except (np.exceptions.AxisError, ValueError): # No detections class_overall = np.zeros(len(class_names)) class_name = "None"