From 2a4585939320e44c5f154744bd4e4abc4019ce0c Mon Sep 17 00:00:00 2001
From: mbsantiago <santiago.mbal@gmail.com>
Date: Wed, 23 Apr 2025 23:19:08 +0100
Subject: [PATCH] Run lint fix

---
 batdetect2/cli/data.py                 | 6 ++++--
 batdetect2/plotting/heatmaps.py        | 1 -
 batdetect2/train/callbacks.py          | 1 -
 batdetect2/train/legacy/train.py       | 2 +-
 batdetect2/train/legacy/train_utils.py | 2 +-
 tests/test_train/test_labels.py        | 1 -
 6 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/batdetect2/cli/data.py b/batdetect2/cli/data.py
index 185476d..94b7dac 100644
--- a/batdetect2/cli/data.py
+++ b/batdetect2/cli/data.py
@@ -35,6 +35,8 @@ def summary(
 ):
     base_dir = base_dir or Path.cwd()
     dataset = load_dataset_from_config(
-        dataset_config, field=field, base_dir=base_dir
+        dataset_config,
+        field=field,
+        base_dir=base_dir,
     )
-    print(f"Number of annotated clips: {len(dataset.clip_annotations)}")
+    print(f"Number of annotated clips: {len(dataset)}")
diff --git a/batdetect2/plotting/heatmaps.py b/batdetect2/plotting/heatmaps.py
index 4927d86..a3df74a 100644
--- a/batdetect2/plotting/heatmaps.py
+++ b/batdetect2/plotting/heatmaps.py
@@ -2,7 +2,6 @@
 
 from typing import Optional, Tuple
 
-import matplotlib.pyplot as plt
 import xarray as xr
 from matplotlib import axes
 
diff --git a/batdetect2/train/callbacks.py b/batdetect2/train/callbacks.py
index 998bbbe..cb486b2 100644
--- a/batdetect2/train/callbacks.py
+++ b/batdetect2/train/callbacks.py
@@ -2,7 +2,6 @@ from lightning import LightningModule, Trainer
 from lightning.pytorch.callbacks import Callback
 from torch.utils.data import DataLoader
 
-from batdetect2.evaluate import match_predictions_and_annotations
 from batdetect2.postprocess import PostprocessorProtocol
 from batdetect2.train.dataset import LabeledDataset, TrainExample
 from batdetect2.types import ModelOutput
diff --git a/batdetect2/train/legacy/train.py b/batdetect2/train/legacy/train.py
index 29fed77..f15fd74 100644
--- a/batdetect2/train/legacy/train.py
+++ b/batdetect2/train/legacy/train.py
@@ -1,4 +1,4 @@
-from typing import Callable, NamedTuple, Optional
+from typing import NamedTuple, Optional
 
 import torch
 from soundevent import data
diff --git a/batdetect2/train/legacy/train_utils.py b/batdetect2/train/legacy/train_utils.py
index 1d0a6bd..b2987d9 100644
--- a/batdetect2/train/legacy/train_utils.py
+++ b/batdetect2/train/legacy/train_utils.py
@@ -1,5 +1,5 @@
-import sys
 import json
+import sys
 from collections import Counter
 from pathlib import Path
 from typing import Dict, Generator, List, Optional, Tuple
diff --git a/tests/test_train/test_labels.py b/tests/test_train/test_labels.py
index 9f55242..91000e4 100644
--- a/tests/test_train/test_labels.py
+++ b/tests/test_train/test_labels.py
@@ -8,7 +8,6 @@ from batdetect2.targets import TargetConfig, TargetProtocol, build_targets
 from batdetect2.targets.rois import ROIConfig
 from batdetect2.targets.terms import TagInfo, TermRegistry
 from batdetect2.train.labels import generate_heatmaps
-from tests.test_targets.test_transform import term_registry
 
 recording = data.Recording(
     samplerate=256_000,