Attend test warnings

This commit is contained in:
mbsantiago 2025-04-22 09:17:17 +01:00
parent c276230bff
commit bfcab0331e
3 changed files with 3 additions and 3 deletions

View File

@ -93,7 +93,7 @@ def convert_xr_dataset_to_raw_prediction(
""" """
detections = [] detections = []
for det_num in range(detection_dataset.dims["detection"]): for det_num in range(detection_dataset.sizes["detection"]):
det_info = detection_dataset.sel(detection=det_num) det_info = detection_dataset.sel(detection=det_num)
geom = geometry_builder( geom = geometry_builder(

View File

@ -8,7 +8,7 @@ from batdetect2.detector import parameters
from batdetect2.utils import audio_utils, detector_utils from batdetect2.utils import audio_utils, detector_utils
@given(duration=st.floats(min_value=0.1, max_value=2)) @given(duration=st.floats(min_value=0.1, max_value=1))
def test_can_compute_correct_spectrogram_width(duration: float): def test_can_compute_correct_spectrogram_width(duration: float):
samplerate = parameters.TARGET_SAMPLERATE_HZ samplerate = parameters.TARGET_SAMPLERATE_HZ
params = parameters.DEFAULT_SPECTROGRAM_PARAMETERS params = parameters.DEFAULT_SPECTROGRAM_PARAMETERS

View File

@ -373,7 +373,7 @@ def test_extract_detection_xr_dataset_empty(
assert isinstance(actual_dataset, xr.Dataset) assert isinstance(actual_dataset, xr.Dataset)
assert "detection" in actual_dataset.dims assert "detection" in actual_dataset.dims
assert actual_dataset.dims["detection"] == 0 assert actual_dataset.sizes["detection"] == 0
assert "scores" in actual_dataset assert "scores" in actual_dataset
assert actual_dataset["scores"].dims == ("detection",) assert actual_dataset["scores"].dims == ("detection",)