fix(preprocessing/video): define VideoFrameU8 and VideoFrameF32 e…

…ven when not type-checking
ruancomelli · Jun 16, 2022 · 453eda1 · 453eda1
1 parent 7f41277
commit 453eda1
Showing 1 changed file with 2 additions and 0 deletions.
diff --git 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