From 24dcc47e73b1c578d4957bc60534c80707e1f0e5 Mon Sep 17 00:00:00 2001 From: mbsantiago Date: Sat, 4 Oct 2025 15:16:11 +0100 Subject: [PATCH] Fix import of DEFAULT_CHECKPOINT_DIR --- src/batdetect2/train/__init__.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/src/batdetect2/train/__init__.py b/src/batdetect2/train/__init__.py index 09d0a88..3a44092 100644 --- a/src/batdetect2/train/__init__.py +++ b/src/batdetect2/train/__init__.py @@ -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",