lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-01-09 16:59:33 +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
from soundevent import data
from batdetect2.models import build_model
from batdetect2.train import FullTrainingConfig, TrainingModule
from batdetect2.config import BatDetect2Config
from batdetect2.train import TrainingModule
from batdetect2.train.train import build_training_module
from batdetect2.typing.preprocess import AudioLoader
def build_default_module():
model = build_model()
config = FullTrainingConfig()
return build_training_module(model, config=config)
config = BatDetect2Config()
return build_training_module(config=config.model_dump())
def test_can_initialize_default_module():