From 453eda162e6ee143e9319c0eac9834e8c9d8764d Mon Sep 17 00:00:00 2001 From: Ruan Comelli Date: Thu, 16 Jun 2022 00:45:44 -0300 Subject: [PATCH] fix(preprocessing/video): define `VideoFrameU8` and `VideoFrameF32` even when not type-checking --- boiling_learning/preprocessing/video.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/boiling_learning/preprocessing/video.py b/boiling_learning/preprocessing/video.py index b4543548..c9b26b1f 100644 --- a/boiling_learning/preprocessing/video.py +++ b/boiling_learning/preprocessing/video.py @@ -24,6 +24,8 @@ VideoFrameF32 = npt.NDArray[np.float32] VideoFrame = Union[VideoFrameU8, VideoFrameF32] else: + VideoFrameU8 = np.ndarray + VideoFrameF32 = np.ndarray VideoFrame = np.ndarray