lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-01-10 00:59:34 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Update test

Browse Source
This commit is contained in:
mbsantiago 2025-10-27 19:39:38 +00:00
parent aa4ad68958
commit 8366410332
1 changed files with 4 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/test_train/test_lightning.py
View File

@ -4,16 +4,15 @@ import lightning as L
import torch import torch
from soundevent import data from soundevent import data
from batdetect2.models import build_model from batdetect2.config import BatDetect2Config
from batdetect2.train import FullTrainingConfig, TrainingModule from batdetect2.train import TrainingModule
from batdetect2.train.train import build_training_module from batdetect2.train.train import build_training_module
from batdetect2.typing.preprocess import AudioLoader from batdetect2.typing.preprocess import AudioLoader
def build_default_module(): def build_default_module():
model = build_model() config = BatDetect2Config()
config = FullTrainingConfig() return build_training_module(config=config.model_dump())
return build_training_module(model, config=config)
def test_can_initialize_default_module(): def test_can_initialize_default_module():