lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2025-06-29 14:41:58 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

address all linting errors from rule NPY201

Browse Source
This commit is contained in:
mbsantiago 2024-11-10 19:13:30 +00:00
parent 95bb0985e7
commit 9e0ad7fd78
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
batdetect2/utils/detector_utils.py
View File

@ -8,6 +8,11 @@ import pandas as pd
import torch
import torch.nn.functional as F
try:
from numpy.exceptions import AxisError
except ImportError:
from numpy import AxisError
import batdetect2.detector.compute_features as feats
import batdetect2.detector.post_process as pp
import batdetect2.utils.audio_utils as au
@ -242,7 +247,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 (AxisError, ValueError):
# No detections
class_overall = np.zeros(len(class_names))
class_name = "None"