Skip to content
This repository was archived by the owner on Jun 10, 2024. It is now read-only.

Commit d8d5d18

Browse files
RomanArzumanyantheHamsta
authored andcommitted
Automated stub generation
1 parent d230bfd commit d8d5d18

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/PyNvCodec/__init__.pyi

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -151,7 +151,8 @@ class CudaBuffer:
151151
def GetNumElems(self) -> int: ...
152152
def GetRawMemSize(self) -> int: ...
153153
def GpuMem(self) -> int: ...
154-
def Make(self, *args, **kwargs) -> Any: ...
154+
@classmethod
155+
def Make(cls, elem_size: int, num_elems: int, gpu_id: int) -> CudaBuffer: ...
155156

156157
class CudaVideoCodec:
157158
__members__: ClassVar[dict] = ... # read-only
@@ -571,7 +572,12 @@ class Surface:
571572
def Format(self) -> PixelFormat: ...
572573
def Height(self, plane: int = ...) -> int: ...
573574
def HostSize(self) -> int: ...
574-
def Make(self, *args, **kwargs) -> Any: ...
575+
@overload
576+
@classmethod
577+
def Make(cls, format: PixelFormat, width: int, height: int, gpu_id: int) -> Surface: ...
578+
@overload
579+
@classmethod
580+
def Make(cls, format: PixelFormat, width: int, height: int, context: int) -> Surface: ...
575581
def NumPlanes(self) -> int: ...
576582
def OwnMemory(self) -> bool: ...
577583
def Pitch(self, plane: int = ...) -> int: ...

0 commit comments

Comments
 (0)