88import warnings
99from concurrent .futures import ThreadPoolExecutor as ThreadPoolExecutorBase
1010from dataclasses import dataclass , field
11- from functools import partial , wraps
11+ from functools import partial
1212from multiprocessing import cpu_count
1313from queue import SimpleQueue
1414from types import MappingProxyType
@@ -374,13 +374,15 @@ def __get__(self, instance: Any, owner: Optional[type] = None) -> Any:
374374
375375
376376@overload
377- def threaded (func : Callable [P , T ]) -> Threaded [P , T ]: ...
377+ def threaded (func : Callable [P , T ]) -> Threaded [P , T ]:
378+ ...
378379
379380
380381@overload
381382def threaded (
382383 func : Callable [P , Generator [T , None , None ]]
383- ) -> Callable [P , IteratorWrapper [P , T ]]: ...
384+ ) -> Callable [P , IteratorWrapper [P , T ]]:
385+ ...
384386
385387
386388def threaded (
@@ -505,15 +507,18 @@ def threaded_iterable(
505507 func : Callable [P , Generator [T , None , None ]],
506508 * ,
507509 max_size : int = 0 ,
508- ) -> "ThreadedIterable[P, T]" : ...
510+ ) -> "ThreadedIterable[P, T]" :
511+ ...
509512
510513
511514@overload
512515def threaded_iterable (
513516 * ,
514517 max_size : int = 0 ,
515518) -> Callable [
516- [Callable [P , Generator [T , None , None ]]], ThreadedIterable [P , T ]]: ...
519+ [Callable [P , Generator [T , None , None ]]], ThreadedIterable [P , T ]
520+ ]:
521+ ...
517522
518523
519524def threaded_iterable (
@@ -522,14 +527,16 @@ def threaded_iterable(
522527 max_size : int = 0 ,
523528) -> Union [
524529 ThreadedIterable [P , T ],
525- Callable [[Callable [P , Generator [T , None , None ]]],
526- ThreadedIterable [P , T ]]
530+ Callable [
531+ [Callable [P , Generator [T , None , None ]]], ThreadedIterable [P , T ]
532+ ]
527533]:
528534 if func is None :
529535 return lambda f : ThreadedIterable (f , max_size = max_size )
530536
531537 return ThreadedIterable (func , max_size = max_size )
532538
539+
533540class IteratorWrapperSeparate (IteratorWrapper ):
534541 def _run (self ) -> Any :
535542 return run_in_new_thread (self ._in_thread )
@@ -550,7 +557,8 @@ def threaded_iterable_separate(
550557 func : Callable [P , Generator [T , None , None ]],
551558 * ,
552559 max_size : int = 0 ,
553- ) -> "ThreadedIterable[P, T]" : ...
560+ ) -> "ThreadedIterable[P, T]" :
561+ ...
554562
555563
556564@overload
@@ -560,7 +568,8 @@ def threaded_iterable_separate(
560568) -> Callable [
561569 [Callable [P , Generator [T , None , None ]]],
562570 ThreadedIterableSeparate [P , T ]
563- ]: ...
571+ ]:
572+ ...
564573
565574
566575def threaded_iterable_separate (
@@ -569,16 +578,16 @@ def threaded_iterable_separate(
569578 max_size : int = 0 ,
570579) -> Union [
571580 ThreadedIterable [P , T ],
572- Callable [[Callable [P , Generator [T , None , None ]]],
573- ThreadedIterableSeparate [P , T ]]
581+ Callable [
582+ [Callable [P , Generator [T , None , None ]]], ThreadedIterableSeparate [P , T ]
583+ ]
574584]:
575585 if func is None :
576586 return lambda f : ThreadedIterableSeparate (f , max_size = max_size )
577587
578588 return ThreadedIterableSeparate (func , max_size = max_size )
579589
580590
581-
582591class CoroutineWaiter :
583592 def __init__ (
584593 self , coroutine : Coroutine [Any , Any , T ],
0 commit comments