feat: Update signature

Author: caffeine-addictt

src/thread/thread.py

@@ -31,6 +31,7 @@ class ParallelProcessing: ...
     _Target_P,
     _Target_T,
     DatasetFunction,
+    Dataset,
     _Dataset_T,
     HookFunction,
     SupportsLength,
@@ -352,12 +353,7 @@ class ParallelProcessing(Generic[_Target_P, _Target_T, _Dataset_T]):
 
     status: ThreadStatus
     function: TargetFunction
-    dataset: Union[
-        Sequence[_Dataset_T],
-        SupportsLength,
-        SupportsGetItem[_Dataset_T],
-        SupportsLengthGetItem[_Dataset_T],
-    ]
+    dataset: Dataset[_Dataset_T]
     max_threads: int
 
     overflow_args: Sequence[Overflow_In]
@@ -367,14 +363,31 @@ class ParallelProcessing(Generic[_Target_P, _Target_T, _Dataset_T]):
     def __init__(
         self,
         function: DatasetFunction[_Dataset_T, _Target_P, _Target_T],
-        dataset: Union[Sequence[_Dataset_T], SupportsLengthGetItem[_Dataset_T]],
+        dataset: Sequence[_Dataset_T],
         max_threads: int = 8,
         *overflow_args: Overflow_In,
         _get_value: Optional[Callable[[Sequence[_Dataset_T], int], _Dataset_T]] = None,
         _length: Optional[Union[int, Callable[[Sequence[_Dataset_T]], int]]] = None,
         **overflow_kwargs: Overflow_In,
     ) -> None: ...
 
+    # Has __len__ and __getitem__ but is not a sequence
+    @overload
+    def __init__(
+        self,
+        function: DatasetFunction[_Dataset_T, _Target_P, _Target_T],
+        dataset: SupportsLengthGetItem[_Dataset_T],
+        max_threads: int = 8,
+        *overflow_args: Overflow_In,
+        _get_value: Optional[
+            Callable[[SupportsLengthGetItem[_Dataset_T], int], _Dataset_T]
+        ] = None,
+        _length: Optional[
+            Union[int, Callable[[SupportsLengthGetItem[_Dataset_T]], int]]
+        ] = None,
+        **overflow_kwargs: Overflow_In,
+    ) -> None: ...
+
     # Has __len__, require _get_value to be set
     @overload
     def __init__(
@@ -383,8 +396,8 @@ def __init__(
         dataset: SupportsLength,
         max_threads: int = 8,
         *overflow_args: Overflow_In,
-        _get_value: Callable[[Sequence[_Dataset_T], int], _Dataset_T],
-        _length: Optional[Union[int, Callable[[Sequence[_Dataset_T]], int]]] = None,
+        _get_value: Callable[[SupportsLength, int], _Dataset_T],
+        _length: Optional[Union[int, Callable[[SupportsLength], int]]] = None,
         **overflow_kwargs: Overflow_In,
     ) -> None: ...
 
@@ -393,27 +406,39 @@ def __init__(
     def __init__(
         self,
         function: DatasetFunction[_Dataset_T, _Target_P, _Target_T],
-        dataset: SupportsLength,
+        dataset: SupportsGetItem[_Dataset_T],
         max_threads: int = 8,
         *overflow_args: Overflow_In,
-        _get_value: Optional[Callable[[Sequence[_Dataset_T], int], _Dataset_T]] = None,
-        _length: Union[int, Callable[[Sequence[_Dataset_T]], int]],
+        _get_value: Optional[
+            Callable[[SupportsGetItem[_Dataset_T], int], _Dataset_T]
+        ] = None,
+        _length: Union[int, Callable[[SupportsGetItem[_Dataset_T]], int]],
         **overflow_kwargs: Overflow_In,
     ) -> None: ...
 
     def __init__(
         self,
         function: DatasetFunction[_Dataset_T, _Target_P, _Target_T],
-        dataset: Union[
-            Sequence[_Dataset_T],
-            SupportsLength,
-            SupportsGetItem[_Dataset_T],
-            SupportsLengthGetItem[_Dataset_T],
-        ],
+        dataset: Dataset[_Dataset_T],
         max_threads: int = 8,
         *overflow_args: Overflow_In,
-        _get_value: Optional[Callable[[Sequence[_Dataset_T], int], _Dataset_T]] = None,
-        _length: Optional[Union[int, Callable[[Sequence[_Dataset_T]], int]]] = None,
+        _get_value: Optional[
+            Union[
+                Callable[[Sequence[_Dataset_T], int], _Dataset_T],
+                Callable[[SupportsGetItem[_Dataset_T], int], _Dataset_T],
+                Callable[[SupportsLength, int], _Dataset_T],
+                Callable[[SupportsLengthGetItem[_Dataset_T], int], _Dataset_T],
+            ]
+        ] = None,
+        _length: Optional[
+            Union[
+                int,
+                Callable[[Sequence[_Dataset_T]], int],
+                Callable[[SupportsGetItem[_Dataset_T]], int],
+                Callable[[SupportsLength], int],
+                Callable[[SupportsLengthGetItem[_Dataset_T]], int],
+            ]
+        ] = None,
         **overflow_kwargs: Overflow_In,
     ) -> None:
         """