From 55f473c9ca06b006b785ed071577a79b53ff7067 Mon Sep 17 00:00:00 2001
From: mbsantiago <santiago.mbal@gmail.com>
Date: Sun, 31 Aug 2025 19:06:09 +0100
Subject: [PATCH] Fix logging issue

---
 src/batdetect2/cli/train.py         | 2 +-
 src/batdetect2/models/bottleneck.py | 1 -
 2 files changed, 1 insertion(+), 2 deletions(-)

diff --git a/src/batdetect2/cli/train.py b/src/batdetect2/cli/train.py
index b16a587..e574b38 100644
--- 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.")
diff --git a/src/batdetect2/models/bottleneck.py b/src/batdetect2/models/bottleneck.py
index 7a38a11..c72f66a 100644
--- 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,