Fix import of DEFAULT_CHECKPOINT_DIR

This commit is contained in:
mbsantiago 2025-10-04 15:16:11 +01:00
parent 160cb6ae30
commit 24dcc47e73

View File

@ -1,3 +1,4 @@
from batdetect2.train.checkpoints import DEFAULT_CHECKPOINT_DIR
from batdetect2.train.config import (
TrainingConfig,
load_train_config,
@ -6,7 +7,7 @@ from batdetect2.train.lightning import (
TrainingModule,
load_model_from_checkpoint,
)
from batdetect2.train.train import DEFAULT_CHECKPOINT_DIR, build_trainer, train
from batdetect2.train.train import build_trainer, train
__all__ = [
"DEFAULT_CHECKPOINT_DIR",