Fix logging issue

2026-01-10 00:59:34 +01:00 · 2025-08-31 19:06:09 +01:00 · 2025-08-31 19:06:09 +01:00 · 55f473c9ca
commit 55f473c9ca
parent 40f6b64611
2 changed files with 1 additions and 2 deletions
--- a/src/batdetect2/cli/train.py
+++ b/src/batdetect2/cli/train.py
@ -70,7 +70,7 @@ def train_command(
        val_annotations = load_dataset_from_config(val_dataset)
        logger.debug(
            "Loaded {num_annotations} validation examples",
-            num_files=len(val_annotations),
+            num_annotations=len(val_annotations),
        )
    else:
        logger.debug("No validation directory provided.")
--- a/src/batdetect2/models/bottleneck.py
+++ b/src/batdetect2/models/bottleneck.py
@ -22,7 +22,6 @@ from torch import nn

 from batdetect2.configs import BaseConfig
 from batdetect2.models.blocks import (
-    LayerConfig,
    SelfAttentionConfig,
    VerticalConv,
    build_layer_from_config,