Improve typing in converter.py

laggykiller · Mar 9, 2024 · 709ad34 · 709ad34
1 parent 603c194
commit 709ad34
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/sticker_convert/converter.py b/src/sticker_convert/converter.py
@@ -635,6 +635,7 @@ def _frames_export_pil(self) -> None:
     def _frames_export_pyav(self) -> None:
         import av  # type: ignore
         from av.container import OutputContainer  # type: ignore
+        from av.video.stream import VideoStream
 
         options = {}
 
@@ -665,6 +666,8 @@ def _frames_export_pyav(self) -> None:
         ) as output:
             output = cast(OutputContainer, output)  # type: ignore
             out_stream = output.add_stream(codec, rate=self.fps, options=options)  # type: ignore
+            out_stream = cast(VideoStream, out_stream)
+            assert isinstance(self.res_w, int) and isinstance(self.res_h, int)
             out_stream.width = self.res_w
             out_stream.height = self.res_h
             out_stream.pix_fmt = pixel_format