fix: error import

williamfzc · Sep 26, 2019 · 665f2ca · 665f2ca
1 parent 738dc00
commit 665f2ca
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/stagesepx/cutter/cut_range.py b/stagesepx/cutter/cut_range.py
@@ -126,7 +126,7 @@ def get_frames(
             out.append(frame)
         return out
 
-    def pick_and_get(self, *args, **kwargs) -> typing.List[toolbox.VideoFrame]:
+    def pick_and_get(self, *args, **kwargs) -> typing.List[VideoFrame]:
         picked = self.pick(*args, **kwargs)
         return self.get_frames(picked, *args, **kwargs)
 

diff --git a/stagesepx/cutter/cut_result.py b/stagesepx/cutter/cut_result.py
@@ -7,7 +7,7 @@
 from loguru import logger
 
 from stagesepx import toolbox
-from stagesepx.video import VideoObject
+from stagesepx.video import VideoObject, VideoFrame
 from stagesepx.cutter.cut_range import VideoCutRange
 
 
@@ -355,8 +355,8 @@ def pick_and_save(
     @staticmethod
     def _prune(
         threshold: float,
-        stages: typing.List[typing.Tuple[str, typing.List[toolbox.VideoFrame]]],
-    ) -> typing.List[typing.Tuple[str, typing.List[toolbox.VideoFrame]]]:
+        stages: typing.List[typing.Tuple[str, typing.List[VideoFrame]]],
+    ) -> typing.List[typing.Tuple[str, typing.List[VideoFrame]]]:
         logger.debug(
             f"start pruning ranges, origin length is {len(stages)}, threshold is {threshold}"
         )