Fixed mypy

Author: DE-AI

av/__init__.py

@@ -18,7 +18,7 @@
 from av.codec.codec import Codec, codecs_available
 from av.codec.context import CodecContext
 from av.codec.hwaccel import HWConfig
-from av.container import open
+from av.container import open, Chapter
 from av.format import ContainerFormat, formats_available
 from av.packet import Packet
 from av.error import *  # noqa: F403; This is limited to exception types.
@@ -41,6 +41,7 @@
     "AudioStream",
     "BitStreamFilterContext",
     "bitstream_filters_available",
+    "Chapter",
     "Codec",
     "codecs_available",
     "CodecContext",

av/container/__init__.py

@@ -1,3 +1,3 @@
-from .core import Container, Flags, open
+from .core import Chapter, Container, Flags, open
 from .input import InputContainer
 from .output import OutputContainer

av/container/core.pyi

@@ -67,7 +67,7 @@ class AudioCodec(IntEnum):
     pcm_u8 = cast(int, ...)
     pcm_vidc = cast(int, ...)
 
-class _Chapter(TypedDict):
+class Chapter(TypedDict):
     id: int
     start: int
     end: int
@@ -102,8 +102,8 @@ class Container:
     ) -> bool: ...
     def set_timeout(self, timeout: Real | None) -> None: ...
     def start_timeout(self) -> None: ...
-    def chapters(self) -> list[_Chapter]: ...
-    def set_chapters(self, chapters: list[_Chapter]) -> None: ...
+    def chapters(self) -> list[Chapter]: ...
+    def set_chapters(self, chapters: list[Chapter]) -> None: ...
 
 @overload
 def open(

tests/test_chapters.py

@@ -1,6 +1,7 @@
 from fractions import Fraction
 
 import av
+from av import Chapter
 
 from .common import fate_suite
 
@@ -42,7 +43,7 @@ def test_chapters() -> None:
 
 
 def test_set_chapters() -> None:
-    chapters = [
+    chapters: list[Chapter] = [
         {
             "id": 1,
             "start": 0,