lionel/batdetect2
1
0
Fork 0
mirror of https://github.com/macaodha/batdetect2.git synced 2026-05-22 22:32:18 +02:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: defer model protocol type imports

Browse Source
This commit is contained in:
mbsantiago 2026-05-06 21:18:25 +01:00
parent cd962fefbd
commit 94746b5f3b
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
src/batdetect2/models/types.py
View File

@ -1,9 +1,10 @@
from typing import Any, NamedTuple, Protocol from typing import TYPE_CHECKING, Any, NamedTuple, Protocol
import torch import torch
from batdetect2.postprocess.types import PostprocessorProtocol if TYPE_CHECKING:
from batdetect2.preprocess.types import PreprocessorProtocol from batdetect2.postprocess.types import PostprocessorProtocol
from batdetect2.preprocess.types import PreprocessorProtocol
__all__ = [ __all__ = [
"BackboneProtocol", "BackboneProtocol",
@ -116,8 +117,8 @@ class DetectorProtocol(ModuleProtocol, Protocol):
class ModelProtocol(ModuleProtocol, Protocol): class ModelProtocol(ModuleProtocol, Protocol):
detector: DetectorProtocol detector: DetectorProtocol
preprocessor: PreprocessorProtocol preprocessor: "PreprocessorProtocol"
postprocessor: PostprocessorProtocol postprocessor: "PostprocessorProtocol"
class_names: list[str] class_names: list[str]
dimension_names: list[str] dimension_names: list[str]