mirror of
https://github.com/macaodha/batdetect2.git
synced 2025-06-29 14:41:58 +02:00
Rename Preprocessor to PreprocessorProtocol
This commit is contained in:
parent
ae6063918c
commit
07f065cf93
@ -1,4 +1,4 @@
|
||||
from typing import Dict, NamedTuple
|
||||
from typing import Dict, NamedTuple, Protocol
|
||||
|
||||
import numpy as np
|
||||
|
||||
@ -15,3 +15,7 @@ class BatDetect2Prediction(NamedTuple):
|
||||
detection_score: float
|
||||
class_scores: Dict[str, float]
|
||||
features: np.ndarray
|
||||
|
||||
|
||||
class PostprocessorProtocol(Protocol):
|
||||
pass
|
||||
|
@ -58,7 +58,7 @@ from batdetect2.preprocess.spectrogram import (
|
||||
)
|
||||
from batdetect2.preprocess.types import (
|
||||
AudioLoader,
|
||||
Preprocessor,
|
||||
PreprocessorProtocol,
|
||||
SpectrogramBuilder,
|
||||
)
|
||||
|
||||
@ -110,7 +110,7 @@ class PreprocessingConfig(BaseConfig):
|
||||
spectrogram: SpectrogramConfig = Field(default_factory=SpectrogramConfig)
|
||||
|
||||
|
||||
class StandardPreprocessor(Preprocessor):
|
||||
class StandardPreprocessor(PreprocessorProtocol):
|
||||
"""Standard implementation of the `Preprocessor` protocol.
|
||||
|
||||
Orchestrates the audio loading and spectrogram generation pipeline using
|
||||
@ -402,7 +402,7 @@ def load_preprocessing_config(
|
||||
|
||||
def build_preprocessor(
|
||||
config: Optional[PreprocessingConfig] = None,
|
||||
) -> Preprocessor:
|
||||
) -> PreprocessorProtocol:
|
||||
"""Factory function to build the standard preprocessor from configuration.
|
||||
|
||||
Creates instances of the required `AudioLoader` and `SpectrogramBuilder`
|
||||
|
@ -158,7 +158,7 @@ class SpectrogramBuilder(Protocol):
|
||||
...
|
||||
|
||||
|
||||
class Preprocessor(Protocol):
|
||||
class PreprocessorProtocol(Protocol):
|
||||
"""Defines a high-level interface for the complete preprocessing pipeline.
|
||||
|
||||
A Preprocessor combines audio loading and spectrogram generation steps.
|
||||
|
Loading…
Reference in New Issue
Block a user