Minor formatting

This commit is contained in:
mbsantiago 2026-03-07 16:09:43 +00:00
parent 4509602e70
commit 8313fe1484
3 changed files with 13 additions and 13 deletions

1
.gitignore vendored
View File

@ -123,3 +123,4 @@ example_data/preprocessed
# Dev notebooks # Dev notebooks
notebooks/tmp notebooks/tmp
/tmp

View File

@ -1,13 +1,14 @@
import sys from typing import (
from typing import Callable, Dict, Generic, Tuple, Type, TypeVar Callable,
Concatenate,
Generic,
ParamSpec,
Type,
TypeVar,
)
from pydantic import BaseModel from pydantic import BaseModel
if sys.version_info >= (3, 10):
from typing import Concatenate, ParamSpec
else:
from typing_extensions import Concatenate, ParamSpec
__all__ = [ __all__ = [
"Registry", "Registry",
"SimpleRegistry", "SimpleRegistry",
@ -45,10 +46,10 @@ class Registry(Generic[T_Type, P_Type]):
def __init__(self, name: str): def __init__(self, name: str):
self._name = name self._name = name
self._registry: Dict[ self._registry: dict[
str, Callable[Concatenate[..., P_Type], T_Type] str, Callable[Concatenate[..., P_Type], T_Type]
] = {} ] = {}
self._config_types: Dict[str, Type[BaseModel]] = {} self._config_types: dict[str, Type[BaseModel]] = {}
def register( def register(
self, self,
@ -74,7 +75,7 @@ class Registry(Generic[T_Type, P_Type]):
return decorator return decorator
def get_config_types(self) -> Tuple[Type[BaseModel], ...]: def get_config_types(self) -> tuple[Type[BaseModel], ...]:
return tuple(self._config_types.values()) return tuple(self._config_types.values())
def get_config_type(self, name: str) -> Type[BaseModel]: def get_config_type(self, name: str) -> Type[BaseModel]:

View File

@ -4,9 +4,7 @@ from pydantic import Field
from soundevent import data from soundevent import data
from batdetect2.core.configs import BaseConfig, load_config from batdetect2.core.configs import BaseConfig, load_config
from batdetect2.evaluate.tasks import ( from batdetect2.evaluate.tasks import TaskConfig
TaskConfig,
)
from batdetect2.evaluate.tasks.classification import ClassificationTaskConfig from batdetect2.evaluate.tasks.classification import ClassificationTaskConfig
from batdetect2.evaluate.tasks.detection import DetectionTaskConfig from batdetect2.evaluate.tasks.detection import DetectionTaskConfig
from batdetect2.logging import CSVLoggerConfig, LoggerConfig from batdetect2.logging import CSVLoggerConfig, LoggerConfig