x

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")