From eb5f5ca262350580174ad8af1c9a87a892657dc9 Mon Sep 17 00:00:00 2001 From: laggykiller Date: Sat, 9 Mar 2024 19:59:03 +0800 Subject: [PATCH] Fix colorspace signature --- av/video/frame.pyi | 5 ++--- av/video/reformatter.pyi | 4 ++-- 2 files changed, 4 insertions(+), 5 deletions(-) diff --git a/av/video/frame.pyi b/av/video/frame.pyi index da6575328..17cacf05a 100644 --- 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 index a68dc9718..abd545332 100644 --- 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,