From 94746b5f3bcad009710acd52a8438167d6e76dd5 Mon Sep 17 00:00:00 2001
From: mbsantiago <santiago.mbal@gmail.com>
Date: Wed, 6 May 2026 21:18:25 +0100
Subject: [PATCH] fix: defer model protocol type imports

---
 src/batdetect2/models/types.py | 11 ++++++-----
 1 file changed, 6 insertions(+), 5 deletions(-)

diff --git a/src/batdetect2/models/types.py b/src/batdetect2/models/types.py
index eb8a181..ca2f236 100644
--- a/src/batdetect2/models/types.py
+++ b/src/batdetect2/models/types.py
@@ -1,9 +1,10 @@
-from typing import Any, NamedTuple, Protocol
+from typing import TYPE_CHECKING, Any, NamedTuple, Protocol
 
 import torch
 
-from batdetect2.postprocess.types import PostprocessorProtocol
-from batdetect2.preprocess.types import PreprocessorProtocol
+if TYPE_CHECKING:
+    from batdetect2.postprocess.types import PostprocessorProtocol
+    from batdetect2.preprocess.types import PreprocessorProtocol
 
 __all__ = [
     "BackboneProtocol",
@@ -116,8 +117,8 @@ class DetectorProtocol(ModuleProtocol, Protocol):
 
 class ModelProtocol(ModuleProtocol, Protocol):
     detector: DetectorProtocol
-    preprocessor: PreprocessorProtocol
-    postprocessor: PostprocessorProtocol
+    preprocessor: "PreprocessorProtocol"
+    postprocessor: "PostprocessorProtocol"
     class_names: list[str]
     dimension_names: list[str]