Run lint fix

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)}")

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
 

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

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

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

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,