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

Minor tweaks

Browse Source
This commit is contained in:
mbsantiago 2025-04-03 16:50:49 +01:00
parent 213b6dfd29
commit 62fa38557e
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
batdetect2/detector/parameters.py
View File

@ -5,7 +5,7 @@ from typing import List, Optional, Union
from pydantic import BaseModel, Field, computed_field from pydantic import BaseModel, Field, computed_field
from batdetect2.train.train_utils import ( from batdetect2.train.legacy.train_utils import (
get_genus_mapping, get_genus_mapping,
get_short_class_names, get_short_class_names,
) )

4
tests/test_models/test_inputs.py
View File

@ -2,7 +2,7 @@ import torch
from hypothesis import given from hypothesis import given
from hypothesis import strategies as st from hypothesis import strategies as st
from batdetect2.models import ModelConfig, ModelType, get_backbone from batdetect2.models import ModelConfig, ModelType, build_architecture
@given( @given(
@ -20,7 +20,7 @@ def test_model_can_process_spectrograms_of_any_width(
input = torch.rand([1, 1, input_height, input_width]) input = torch.rand([1, 1, input_height, input_width])
model = get_backbone( model = build_architecture(
config=ModelConfig( config=ModelConfig(
name=model_type, # type: ignore name=model_type, # type: ignore
input_height=input_height, input_height=input_height,