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
|
# Dev notebooks
|
||||||
notebooks/tmp
|
notebooks/tmp
|
||||||
|
/tmp
|
||||||
|
|||||||
@ -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]:
|
||||||
|
|||||||
@ -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
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user