tensorneko-util 0.3.21__py3-none-any.whl → 0.3.22__py3-none-any.whl

tensorneko_util/io/audio/audio_writer.pyi

@@ -0,0 +1,29 @@
+from typing import overload, Union
+from pathlib import Path
+
+from .audio_data import AudioData
+from ...backend.audio_lib import AudioLib
+from ...util.type import T_ARRAY
+
+
+class AudioWriter:
+
+    @classmethod
+    @overload
+    def to(cls, path: Union[str, Path], audio: AudioData, channel_first: bool = True, backend: AudioLib = None
+    ) -> None: ...
+
+    @classmethod
+    @overload
+    def to(cls, path: Union[str, Path], audio: T_ARRAY, sample_rate: int = 16000, channel_first: bool = True,
+        backend: AudioLib = None
+    ): ...
+
+    @overload
+    def __new__(cls, path: Union[str, Path], audio: AudioData, channel_first: bool = True, backend: AudioLib = None
+    ) -> None: ...
+
+    @overload
+    def __new__(cls, path: Union[str, Path], audio: T_ARRAY, sample_rate: int = 16000, channel_first: bool = True,
+        backend: AudioLib = None
+    ): ...

tensorneko_util/io/video/video_reader.pyi

@@ -0,0 +1,38 @@
+from typing import Optional, overload, Union
+from pathlib import Path
+
+from numpy import ndarray
+
+from .video_data import VideoData
+from ...backend.visual_lib import VisualLib
+
+
+class VideoReader:
+
+    @classmethod
+    def of(cls, path: Union[str, Path], channel_first: bool = True, backend: Optional[VisualLib] = None) -> VideoData:
+        ...
+
+    @classmethod
+    def with_indexes(cls, path: Union[str, Path], indexes: ndarray,
+        channel_first: bool = True, backend: Optional[VisualLib] = None
+    ) -> VideoData:
+        ...
+
+    @classmethod
+    @overload
+    def with_range(cls, path: Union[str, Path], start: int, end: int, step: int, channel_first: bool = True,
+        backend: Optional[VisualLib] = None
+    ) -> VideoData:
+        ...
+
+    @classmethod
+    @overload
+    def with_range(cls, path: Union[str, Path], end: int, channel_first: bool = True,
+        backend: Optional[VisualLib] = None
+    ) -> VideoData:
+        ...
+
+    def __new__(cls, path: Union[str, Path], channel_first: bool = True, backend: Optional[VisualLib] = None
+    ) -> VideoData:
+        ...

tensorneko_util/io/video/video_writer.pyi

@@ -0,0 +1,33 @@
+from typing import overload, Optional, Union
+from pathlib import Path
+
+from .video_data import VideoData
+from ...backend.visual_lib import VisualLib
+from ...util.type import T_ARRAY
+
+
+class VideoWriter:
+
+    @classmethod
+    @overload
+    def to(cls, path: Union[str, Path], video: VideoData, audio_codec: str = None, channel_first: bool = False,
+        backend: VisualLib = None
+    ) -> None: ...
+
+    @classmethod
+    @overload
+    def to(cls, path: Union[str, Path], video: T_ARRAY, video_fps: float, audio: T_ARRAY = None,
+        audio_fps: Optional[int] = None, audio_codec: str = None, channel_first: bool = False,
+        backend: VisualLib = None
+    ) -> None: ...
+
+    @overload
+    def __new__(cls, path: Union[str, Path], video: VideoData, audio_codec: str = None, channel_first: bool = False,
+        backend: VisualLib = None
+    ) -> None: ...
+
+    @overload
+    def __new__(cls, path: Union[str, Path], video: T_ARRAY, video_fps: float, audio: T_ARRAY = None,
+        audio_fps: Optional[int] = None, audio_codec: str = None, channel_first: bool = False,
+        backend: VisualLib = None
+    ) -> None: ...

tensorneko_util/preprocess/crop.pyi

@@ -0,0 +1,15 @@
+from typing import overload, Union
+
+from numpy import ndarray
+
+
+@overload
+def crop_with_padding(image: ndarray, x1: int, x2: int, y1: int, y2: int, pad_value: Union[int, float] = 0.,
+    batch: bool = False
+) -> ndarray: ...
+
+
+@overload
+def crop_with_padding(image: ndarray, x1: ndarray, x2: ndarray, y1: ndarray, y2: ndarray,
+    pad_value: Union[int, float] = 0., batch: bool = False
+) -> ndarray: ...

tensorneko_util/util/dispatched_misc.pyi

@@ -0,0 +1,23 @@
+from typing import overload, List
+
+from numpy import ndarray
+
+
+@overload
+def sparse2binary(x: ndarray, length: int = None) -> ndarray:
+    ...
+
+
+@overload
+def sparse2binary(x: List[int], length: int = None) -> List[int]:
+    ...
+
+
+@overload
+def binary2sparse(x: ndarray) -> ndarray:
+    ...
+
+
+@overload
+def binary2sparse(x: List[int]) -> List[int]:
+    ...

tensorneko_util/util/ref.pyi

@@ -0,0 +1,38 @@
+from __future__ import annotations
+
+from typing import Generic, overload
+
+from .type import P
+
+
+class Ref(Generic[P]):
+    value: P
+    @overload
+    def __new__(cls, value: str) -> StringRef: ...
+    @overload
+    def __new__(cls, value: int) -> IntRef: ...
+    @overload
+    def __new__(cls, value: float) -> FloatRef: ...
+    @overload
+    def __new__(cls, value: bool) -> BoolRef: ...
+
+class StringRef(Ref[str]):
+    def __str__(self) -> str: ...
+
+class IntRef(Ref[int]):
+    def __int__(self) -> int: ...
+
+class FloatRef(Ref[float]):
+    def __float__(self) -> float: ...
+
+class BoolRef(Ref[bool]):
+    def __bool__(self) -> bool: ...
+
+@overload
+def ref(value: str) -> StringRef: ...
+@overload
+def ref(value: int) -> IntRef: ...
+@overload
+def ref(value: float) -> FloatRef: ...
+@overload
+def ref(value: bool) -> BoolRef: ...

tensorneko_util/version.txt

@@ -1 +1 @@
-0.3.21
+0.3.22