From 9f8747512917a74a90b6e6c4def04b7e78765995 Mon Sep 17 00:00:00 2001 From: Avasam Date: Tue, 4 Jul 2023 16:44:32 -0400 Subject: [PATCH] Fix partially unknown Mat --- .../core/misc/python/package/mat_wrapper/__init__.py | 10 ++++++++-- 1 file changed, 8 insertions(+), 2 deletions(-) diff --git a/modules/core/misc/python/package/mat_wrapper/__init__.py b/modules/core/misc/python/package/mat_wrapper/__init__.py index 7309c32b01..44832e7109 100644 --- a/modules/core/misc/python/package/mat_wrapper/__init__.py +++ b/modules/core/misc/python/package/mat_wrapper/__init__.py @@ -1,12 +1,18 @@ __all__ = [] -import sys import numpy as np import cv2 as cv +from typing import TYPE_CHECKING, Any + +# Type subscription is not possible in python 3.8 +if TYPE_CHECKING: + _NDArray = np.ndarray[Any, np.dtype[np.generic]] +else: + _NDArray = np.ndarray # NumPy documentation: https://numpy.org/doc/stable/user/basics.subclassing.html -class Mat(np.ndarray): +class Mat(_NDArray): ''' cv.Mat wrapper for numpy array.