[bugfix] Fix incorrect type hint leading to runtime failures #468

Breakthrough · Dec 23, 2024 · 69c1922 · 69c1922
1 parent d7b0972
commit 69c1922
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/scenedetect/__init__.py b/scenedetect/__init__.py
@@ -56,7 +56,7 @@
 
 # Used for module identification and when printing version & about info
 # (e.g. calling `scenedetect version` or `scenedetect about`).
-__version__ = "0.6.5.1"
+__version__ = "0.6.5.2"
 
 init_logger()
 logger = getLogger("pyscenedetect")

diff --git a/scenedetect/scene_manager.py b/scenedetect/scene_manager.py
@@ -410,13 +410,13 @@ def write_scene_list_html(
 
 
 def _scale_image(
-    image: cv2.Mat,
+    image: np.ndarray,
     aspect_ratio: float,
     height: ty.Optional[int],
     width: ty.Optional[int],
     scale: ty.Optional[float],
     interpolation: Interpolation,
-) -> cv2.Mat:
+) -> np.ndarray:
     # TODO: Combine this resize with the ones below.
     if aspect_ratio is not None:
         image = cv2.resize(
@@ -709,9 +709,9 @@ def generate_timecode_list(self, scene_list: SceneList) -> ty.List[ty.Iterable[F
 
     def resize_image(
         self,
-        image: cv2.Mat,
+        image: np.ndarray,
         aspect_ratio: float,
-    ) -> cv2.Mat:
+    ) -> np.ndarray:
         return _scale_image(
             image, aspect_ratio, self._height, self._width, self._scale, self._interpolation
         )