Add annotations to encode_frames_with_qminmax

Author: WyattBlue

av/stream.pyi

@@ -48,6 +48,9 @@ class Stream:
     language: str | None
     type: Literal["video", "audio", "data", "subtitle", "attachment"]
 
+    # From context
+    codec_tag: str
+
 class DataStream(Stream):
     type: Literal["data"]
     name: str | None

tests/test_encode.py

@@ -1,3 +1,5 @@
+from __future__ import annotations
+
 import io
 import math
 from fractions import Fraction
@@ -409,7 +411,9 @@ def test_max_b_frames(self) -> None:
             assert actual_max_b_frames <= max_b_frames
 
 
-def encode_frames_with_qminmax(frames: list, shape: tuple, qminmax: tuple) -> int:
+def encode_frames_with_qminmax(
+    frames: list[VideoFrame], shape: tuple[int, int, int], qminmax: tuple[int, int]
+) -> int:
     """
     Encode a video with the given quantiser limits, and return how many enocded
     bytes we made in total.