x

2025-03-03 12:03:37 +08:00
parent 25a66cd95c
commit f0f71f7f81
6 changed files with 448 additions and 12 deletions
--- a/app/_typing/init.py
+++ b/app/_typing/init.py
@ -1,7 +1,13 @@
 from typing import TypeVar
-import numpy as np
-from jaxtyping import Float, Int, Bool, Num

-T = TypeVar("T")
+import numpy as np
+from jaxtyping import Array as JaxArray
+from jaxtyping import Bool, Float, Int, Num
+
+try:
+    from cv2.typing import MatLike
+except ImportError:
+    MatLike = np.ndarray

 NDArray = np.ndarray
+T = TypeVar("T")