@@ -362,19 +362,7 @@ class ParallelProcessing(Generic[_Target_P, _Target_T, _Dataset_T]):
362362 overflow_args : Sequence [Overflow_In ]
363363 overflow_kwargs : Mapping [str , Overflow_In ]
364364
365- @overload
366- def __init__ (
367- self ,
368- function : DatasetFunction [_Dataset_T , _Target_P , _Target_T ],
369- dataset : Sequence [_Dataset_T ],
370- max_threads : int = 8 ,
371- * overflow_args : Overflow_In ,
372- _get_value : Optional [Callable [[Sequence [_Dataset_T ], int ], _Dataset_T ]] = None ,
373- _length : Optional [Union [int , Callable [[Sequence [_Dataset_T ]], int ]]] = None ,
374- ** overflow_kwargs : Overflow_In ,
375- ) -> None : ...
376-
377- # Has __len__ and __getitem__ but is not a sequence
365+ # Has __len__ and __getitem__
378366 @overload
379367 def __init__ (
380368 self ,
@@ -431,19 +419,15 @@ def __init__(
431419 * overflow_args : Overflow_In ,
432420 _get_value : Optional [
433421 Union [
434- Callable [[Sequence [_Dataset_T ], int ], _Dataset_T ],
422+ Callable [[SupportsLengthGetItem [_Dataset_T ], int ], _Dataset_T ],
435423 Callable [[SupportsGetItem [_Dataset_T ], int ], _Dataset_T ],
436424 Callable [[SupportsLength , int ], _Dataset_T ],
437- Callable [[SupportsLengthGetItem [_Dataset_T ], int ], _Dataset_T ],
438425 ]
439426 ] = None ,
440427 _length : Optional [
441428 Union [
442429 int ,
443- Callable [[Sequence [_Dataset_T ]], int ],
444- Callable [[SupportsGetItem [_Dataset_T ]], int ],
445- Callable [[SupportsLength ], int ],
446- Callable [[SupportsLengthGetItem [_Dataset_T ]], int ],
430+ Callable [[Any ], int ],
447431 ]
448432 ] = None ,
449433 ** overflow_kwargs : Overflow_In ,
0 commit comments