Fix colorspace signature

PyAV-Org · Mar 9, 2024 · bd6104a · bd6104a
1 parent 9ad1d91
commit bd6104a
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 5 deletions.
diff --git a/av/video/frame.pyi b/av/video/frame.pyi
@@ -8,7 +8,6 @@ from av.frame import Frame
 
 from .format import VideoFormat
 from .plane import VideoPlane
-from .reformatter import ColorRange, Colorspace
 
 _SupportedNDarray = Union[
     np.ndarray[Any, np.dtype[np.uint8]],
@@ -47,8 +46,8 @@ class VideoFrame(Frame):
         width: int | None = None,
         height: int | None = None,
         format: str | None = None,
-        src_colorspace: Colorspace | None = None,
-        dst_colorspace: Colorspace | None = None,
+        src_colorspace: int | None = None,
+        dst_colorspace: int | None = None,
         interpolation: int | str | None = None,
         src_color_range: int | str | None = None,
         dst_color_range: int | str | None = None,

diff --git a/av/video/reformatter.pyi b/av/video/reformatter.pyi
@@ -43,8 +43,8 @@ class VideoReformatter:
         width: int | None = None,
         height: int | None = None,
         format: str | None = None,
-        src_colorspace: Colorspace | None = None,
-        dst_colorspace: Colorspace | None = None,
+        src_colorspace: int | None = None,
+        dst_colorspace: int | None = None,
         interpolation: int | str | None = None,
         src_color_range: int | str | None = None,
         dst_color_range: int | str | None = None,