mirror of
https://github.com/macaodha/batdetect2.git
synced 2026-04-04 15:20:19 +02:00
Minor formatting
This commit is contained in:
parent
4509602e70
commit
8313fe1484
1
.gitignore
vendored
1
.gitignore
vendored
@ -123,3 +123,4 @@ example_data/preprocessed
|
||||
|
||||
# Dev notebooks
|
||||
notebooks/tmp
|
||||
/tmp
|
||||
|
||||
@ -1,13 +1,14 @@
|
||||
import sys
|
||||
from typing import Callable, Dict, Generic, Tuple, Type, TypeVar
|
||||
from typing import (
|
||||
Callable,
|
||||
Concatenate,
|
||||
Generic,
|
||||
ParamSpec,
|
||||
Type,
|
||||
TypeVar,
|
||||
)
|
||||
|
||||
from pydantic import BaseModel
|
||||
|
||||
if sys.version_info >= (3, 10):
|
||||
from typing import Concatenate, ParamSpec
|
||||
else:
|
||||
from typing_extensions import Concatenate, ParamSpec
|
||||
|
||||
__all__ = [
|
||||
"Registry",
|
||||
"SimpleRegistry",
|
||||
@ -45,10 +46,10 @@ class Registry(Generic[T_Type, P_Type]):
|
||||
|
||||
def __init__(self, name: str):
|
||||
self._name = name
|
||||
self._registry: Dict[
|
||||
self._registry: dict[
|
||||
str, Callable[Concatenate[..., P_Type], T_Type]
|
||||
] = {}
|
||||
self._config_types: Dict[str, Type[BaseModel]] = {}
|
||||
self._config_types: dict[str, Type[BaseModel]] = {}
|
||||
|
||||
def register(
|
||||
self,
|
||||
@ -74,7 +75,7 @@ class Registry(Generic[T_Type, P_Type]):
|
||||
|
||||
return decorator
|
||||
|
||||
def get_config_types(self) -> Tuple[Type[BaseModel], ...]:
|
||||
def get_config_types(self) -> tuple[Type[BaseModel], ...]:
|
||||
return tuple(self._config_types.values())
|
||||
|
||||
def get_config_type(self, name: str) -> Type[BaseModel]:
|
||||
|
||||
@ -4,9 +4,7 @@ from pydantic import Field
|
||||
from soundevent import data
|
||||
|
||||
from batdetect2.core.configs import BaseConfig, load_config
|
||||
from batdetect2.evaluate.tasks import (
|
||||
TaskConfig,
|
||||
)
|
||||
from batdetect2.evaluate.tasks import TaskConfig
|
||||
from batdetect2.evaluate.tasks.classification import ClassificationTaskConfig
|
||||
from batdetect2.evaluate.tasks.detection import DetectionTaskConfig
|
||||
from batdetect2.logging import CSVLoggerConfig, LoggerConfig
|
||||
|
||||
Loading…
Reference in New Issue
Block a user