tensorneko-util 0.3.21__tar.gz → 0.3.23__tar.gz

tensorneko_util-0.3.23/LICENSE

@@ -0,0 +1,21 @@
+MIT License
+
+Copyright (c) 2021 ControlNet
+
+Permission is hereby granted, free of charge, to any person obtaining a copy
+of this software and associated documentation files (the "Software"), to deal
+in the Software without restriction, including without limitation the rights
+to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
+copies of the Software, and to permit persons to whom the Software is
+furnished to do so, subject to the following conditions:
+
+The above copyright notice and this permission notice shall be included in all
+copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
+IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
+FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
+AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
+LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
+OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
+SOFTWARE.

{tensorneko_util-0.3.21 → tensorneko_util-0.3.23}/PKG-INFO

@@ -1,18 +1,13 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: tensorneko_util
-Version: 0.3.21
+Version: 0.3.23
 Summary: The Utils for Library TensorNeko.
 Home-page: https://github.com/ControlNet/tensorneko
 Author: ControlNet
 Author-email: smczx@hotmail.com
-License: UNKNOWN
 Project-URL: Bug Tracker, https://github.com/ControlNet/tensorneko/issues
 Project-URL: Source Code, https://github.com/ControlNet/tensorneko
-Description: 
-        The independent util library for TensorNeko. This library doesn't require PyTorch as a dependency.
-        
 Keywords: deep learning,pytorch,AI,data processing
-Platform: UNKNOWN
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.7
 Classifier: Programming Language :: Python :: 3.8
@@ -20,6 +15,8 @@ Classifier: Programming Language :: Python :: 3.9
 Classifier: Programming Language :: Python :: 3.10
 Classifier: Programming Language :: Python :: 3.11
 Classifier: Programming Language :: Python :: 3.12
+Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Operating System :: OS Independent
 Classifier: Intended Audience :: Developers
@@ -27,3 +24,21 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: Topic :: Utilities
 Requires-Python: >=3.7
 Description-Content-Type: text/markdown
+License-File: LICENSE
+Requires-Dist: numpy>=1.20.1
+Requires-Dist: einops>=0.3.0
+Dynamic: author
+Dynamic: author-email
+Dynamic: classifier
+Dynamic: description
+Dynamic: description-content-type
+Dynamic: home-page
+Dynamic: keywords
+Dynamic: license-file
+Dynamic: project-url
+Dynamic: requires-dist
+Dynamic: requires-python
+Dynamic: summary
+
+
+The independent util library for TensorNeko. This library doesn't require PyTorch as a dependency.

{tensorneko_util-0.3.21 → tensorneko_util-0.3.23}/README.md

@@ -32,7 +32,7 @@ pip install tensorneko  # for PyTorch only
 pip install tensorneko[lightning]  # for PyTorch and Lightning
 ```
 
-To use the library without PyTorch and PyTorch Lightning, you can install the util library (support Python 3.7 ~ 3.12 with limited features) with following command.
+To use the library without PyTorch and PyTorch Lightning, you can install the util library (support Python 3.7 ~ 3.14 with limited features) with following command.
 ```shell
 pip install tensorneko_util
 ```

{tensorneko_util-0.3.21 → tensorneko_util-0.3.23}/setup.py

@@ -51,6 +51,7 @@ setuptools.setup(
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
         "Programming Language :: Python :: 3.12",
+        "Programming Language :: Python :: 3.13",
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
         "Intended Audience :: Developers",

{tensorneko_util-0.3.21 → tensorneko_util-0.3.23}/setup_util.py

@@ -47,6 +47,8 @@ setuptools.setup(
         "Programming Language :: Python :: 3.10",
         "Programming Language :: Python :: 3.11",
         "Programming Language :: Python :: 3.12",
+        "Programming Language :: Python :: 3.13",
+        "Programming Language :: Python :: 3.14",
         "License :: OSI Approved :: MIT License",
         "Operating System :: OS Independent",
         "Intended Audience :: Developers",

tensorneko_util-0.3.23/src/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-0.3.23/src/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-0.3.23/src/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-0.3.23/src/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-0.3.23/src/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-0.3.21 → tensorneko_util-0.3.23}/src/tensorneko_util/util/dispatcher.py

@@ -1,12 +1,28 @@
 from __future__ import annotations
 
 import inspect
+import sys
 import warnings
 from typing import Callable, Dict, List, Generic, Sequence, Optional, TYPE_CHECKING, overload
 
 from .type import T
 
 
+def _is_union_type(annotation) -> bool:
+    """
+    Check if the annotation is a Union type.
+    Compatible with Python 3.8 ~ 3.14+
+    """
+    if sys.version_info >= (3, 14):
+        # Python 3.14+: str(Union[int, str]) returns "int | str"
+        # Use typing.get_origin() for reliable detection
+        from typing import Union, get_origin
+        return get_origin(annotation) is Union
+    else:
+        # Python < 3.14: str(Union[int, str]) returns "typing.Union[int, str]"
+        return str(annotation)[:13] == "typing.Union["
+
+
 class DispatcherTypeWarning(Warning):
     pass
 
@@ -38,8 +54,7 @@ class Dispatcher:
             possible_types = [[]]
 
             for param in parameters:
-                # if is union type
-                is_union = str(param.annotation)[:13] == "typing.Union["
+                is_union = _is_union_type(param.annotation)
 
                 # if no default value and not union, just append to each possible types
                 if param.default is inspect.Signature.empty and not is_union:

tensorneko_util-0.3.23/src/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-0.3.23/src/tensorneko_util/version.txt

@@ -0,0 +1 @@
+0.3.23