Include a few more definitions

WyattBlue · Nov 29, 2023 · cf8e1fd · cf8e1fd
1 parent d0b06d6
commit cf8e1fd
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 2 deletions.
diff --git a/av/__init__.pyi b/av/__init__.pyi
@@ -7,3 +7,6 @@ from .codec.codec import *
 from .container.core import *
 from .container.input import InputContainer
 from .container.output import OutputContainer
+from .error import *
+from .video.format import *
+from .video.frame import *
diff --git a/av/container/input.pyi b/av/container/input.pyi
@@ -1,12 +1,16 @@
+from av.frame import Frame
+from av.packet import Packet
+
 from .core import Container
 from .streams import Stream
 
 class InputContainer(Container):
     bit_rate: int
     size: int
 
-    def demux(self, *args, **kwargs): ...
-    def decode(self, *args, **kwargs): ...
+    def close(self): ...
+    def demux(self, *args, **kwargs) -> Packet: ...
+    def decode(self, *args, **kwargs) -> Frame: ...
     def seek(
         self,
         offset: int,

diff --git a/av/error.pyi b/av/error.pyi
@@ -13,3 +13,4 @@ class LookupError(FFmpegError): ...
 class HTTPError(FFmpegError): ...
 class HTTPClientError(FFmpegError): ...
 class UndefinedError(FFmpegError): ...
+class InvalidDataError(ValueError): ...