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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: tensorneko_util
-Version: 0.3.22
+Version: 0.3.23
 Summary: The Utils for Library TensorNeko.
 Home-page: https://github.com/ControlNet/tensorneko
 Author: ControlNet
@@ -16,6 +16,7 @@ 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

{tensorneko_util-0.3.22 → 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.13 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.22 → tensorneko_util-0.3.23}/setup_util.py

@@ -48,6 +48,7 @@ setuptools.setup(
         "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.22 → 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/version.txt

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