tensorneko-util 0.3.18__tar.gz → 0.3.20__tar.gz

{tensorneko_util-0.3.18 → tensorneko_util-0.3.20}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: tensorneko_util
-Version: 0.3.18
+Version: 0.3.20
 Summary: The Utils for Library TensorNeko.
 Home-page: https://github.com/ControlNet/tensorneko
 Author: ControlNet

{tensorneko_util-0.3.18 → tensorneko_util-0.3.20}/README.md

@@ -12,7 +12,7 @@
 <div align="center">    
     <a href="https://www.python.org/"><img src="https://img.shields.io/pypi/pyversions/tensorneko?style=flat-square"></a>
     <a href="https://pytorch.org/"><img src="https://img.shields.io/badge/PyTorch-%3E%3D1.9.0-EE4C2C?style=flat-square&logo=pytorch"></a>
-    <a href="https://www.pytorchlightning.ai/"><img src="https://img.shields.io/badge/Lightning-2.0.*%20|%202.1.*%20|%202.2.*-792EE5?style=flat-square&logo=lightning"></a>
+    <a href="https://www.pytorchlightning.ai/"><img src="https://img.shields.io/badge/Lightning-2.*-792EE5?style=flat-square&logo=lightning"></a>
 </div>
 
 <div align="center">

{tensorneko_util-0.3.18 → tensorneko_util-0.3.20}/src/tensorneko_util/io/image/image_reader.py

@@ -35,7 +35,10 @@ class ImageReader:
             if not VisualLib.opencv_available():
                 raise ValueError("OpenCV is not installed.")
             import cv2
-            img = cv2.imread(path).astype(np.float32) / 255.0
+            img = cv2.imread(path)
+            if img is None:
+                raise ValueError("Invalid image path: {}".format(path))
+            img = img.astype(np.float32) / 255.0
             img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
         elif backend == VisualLib.MATPLOTLIB:
             if not VisualLib.matplotlib_available():

{tensorneko_util-0.3.18 → tensorneko_util-0.3.20}/src/tensorneko_util/io/json/json_writer.py

@@ -9,7 +9,7 @@ from .._path_conversion import _path2str
 class JsonWriter:
 
     @staticmethod
-    def _write_json(path: str, obj: Union[dict, list], encoding: str = "UTF-8", indent: int = 4,
+    def _write_json(path: str, obj: Union[dict, list], encoding: str = "UTF-8", indent: int = 2,
         ensure_ascii: bool = False, fast: bool = True
     ) -> None:
         if not fast:
@@ -28,7 +28,7 @@ class JsonWriter:
                 file.write(orjson.dumps(obj, option=option))
 
     @classmethod
-    def to(cls, path: Union[str, Path], obj: Union[dict, list, object], encoding: str = "UTF-8", indent: int = 4,
+    def to(cls, path: Union[str, Path], obj: Union[dict, list, object], encoding: str = "UTF-8", indent: int = 2,
         ensure_ascii: bool = False, fast: bool = True
     ) -> None:
         """
@@ -68,7 +68,7 @@ class JsonWriter:
         else:
             raise TypeError("Not implemented type. Only support dict, list, json_data.")
 
-    def __new__(cls, path: Union[str, Path], obj: Union[dict, list, object], encoding: str = "UTF-8", indent: int = 4,
+    def __new__(cls, path: Union[str, Path], obj: Union[dict, list, object], encoding: str = "UTF-8", indent: int = 2,
         ensure_ascii: bool = False, fast: bool = True
     ) -> None:
         """Alias of :meth:`~tensorneko.io.json.json_writer.JsonWriter.to`."""

{tensorneko_util-0.3.18 → tensorneko_util-0.3.20}/src/tensorneko_util/io/pickle/pickle_reader.py

@@ -1,5 +1,5 @@
 import pickle
-from typing import Union
+from typing import Union, Any
 from pathlib import Path
 
 from .._path_conversion import _path2str
@@ -8,7 +8,7 @@ from .._path_conversion import _path2str
 class PickleReader:
 
     @classmethod
-    def of(cls, path: Union[str, Path]) -> None:
+    def of(cls, path: Union[str, Path]) -> Any:
         """
         Save the object to a file.
 
@@ -19,7 +19,7 @@ class PickleReader:
         with open(path, 'rb') as f:
             return pickle.load(f)
 
-    def __new__(cls, path: Union[str, Path]) -> None:
+    def __new__(cls, path: Union[str, Path]) -> Any:
         """Alias to :meth:`~tensorneko_util.io.pickle.PickleReader.of`."""
         path = _path2str(path)
         return cls.of(path)

{tensorneko_util-0.3.18 → tensorneko_util-0.3.20}/src/tensorneko_util/util/__init__.py

@@ -12,6 +12,7 @@ from .eventbus import Event, EventBus, EventHandler, subscribe
 from .singleton import Singleton
 from .downloader import download_file, download_file_thread, download_files_thread
 from .window_merger import WindowMerger
+from .multi_layer_indexer import MultiLayerIndexer
 
 tensorneko_util_path = get_tensorneko_util_path()
 
@@ -56,5 +57,6 @@ __all__ = [
     "download_file_thread",
     "download_files_thread",
     "WindowMerger",
-    "Registry"
+    "Registry",
+    "MultiLayerIndexer"
 ]

tensorneko_util-0.3.20/src/tensorneko_util/util/multi_layer_indexer.py

@@ -0,0 +1,113 @@
+from typing import Union, List
+
+T_Counts = Union[int, List["Counts"]]
+
+
+class MultiLayerIndexer:
+    """
+    Multi-layer indexer for hierarchical indexing structures.
+
+    Args:
+        counts (``int`` | ``List[int | List]``): A nested structure representing the counts at each layer.
+            It can be an integer or a list of counts (which can themselves be lists).
+
+    Examples::
+
+        # Test Case 1
+        counts = [5, 10, 15]
+        indexer = MultiLayerIndexer(counts)
+        print(len(indexer), indexer(12))  # Output: 30 [1, 7]
+
+        # Test Case 2
+        counts = [[3, 2], 4, [1, 1, 1]]
+        indexer = MultiLayerIndexer(counts)
+        print(len(indexer), indexer(8))  # Output: 12 [1, 3]
+
+        # Test Case 3 (Error Handling)
+        try:
+            counts = [2, [3, [4, 5]]]
+            indexer = MultiLayerIndexer(counts)
+            print(len(indexer), indexer(20))  # Should raise IndexError
+        except IndexError as e:
+            print(e)  # Output: Index out of bounds
+
+    """
+
+    def __init__(self, counts: T_Counts):
+        self._counts = counts
+        self._total = self._total_counts(counts)
+
+    def __call__(self, index: int) -> List[int]:
+        """
+        Given a global index, return the indices at each layer.
+
+        Args:
+            index (``int``): The global index.
+
+        Returns:
+            ``List[int]``: A list of indices at each layer corresponding to the global index.
+
+        Raises:
+            ``IndexError``: If the provided global index is out of bounds.
+        """
+        return self._multi_layer_index(self._counts, index)
+
+    @classmethod
+    def _multi_layer_index(cls, counts: T_Counts, index: int) -> List[int]:
+        """
+        A recursive helper function to compute the indices at each layer.
+
+        Args:
+            counts (``int`` | ``List[int | List]``): The counts at the current layer.
+            index (``int``): The index at the current layer.
+
+        Returns:
+            ``list``: A list of indices at each layer.
+
+        Raises:
+            ``IndexError``: If the provided index exceeds the available counts.
+        """
+        if isinstance(counts, int):
+            if index < counts:
+                return [index]
+            else:
+                raise IndexError("Index out of bounds")
+        elif isinstance(counts, list):
+            cumulative_counts = [0]
+            for c in counts:
+                total_c = cls._total_counts(c)
+                cumulative_counts.append(cumulative_counts[-1] + total_c)
+            for i in range(len(cumulative_counts) - 1):
+                if cumulative_counts[i] <= index < cumulative_counts[i + 1]:
+                    sub_idx = index - cumulative_counts[i]
+                    sub_indices = cls._multi_layer_index(counts[i], sub_idx)
+                    return [i] + sub_indices
+            raise IndexError("Index out of bounds")
+        else:
+            raise ValueError("Invalid counts structure")
+
+    @classmethod
+    def _total_counts(cls, counts: T_Counts) -> int:
+        """
+        Compute the total counts of items under the given counts structure.
+
+        Args:
+            counts (``int`` | ``List[int | List]``): The counts structure.
+
+        Returns:
+            ``int``: The total count of items.
+
+        Raises:
+            ``ValueError``: If the counts structure is invalid.
+        """
+        if isinstance(counts, int):
+            if counts < 0:
+                raise ValueError("Counts cannot be negative")
+            return counts
+        elif isinstance(counts, list):
+            return sum(cls._total_counts(c) for c in counts)
+        else:
+            raise ValueError("Invalid counts structure")
+
+    def __len__(self) -> int:
+        return self._total

tensorneko_util-0.3.20/src/tensorneko_util/version.txt

@@ -0,0 +1 @@
+0.3.20