ConcurrentMappingIterable: fix generic typing

ebonnal · Sep 28, 2024 · 3856834 · 3856834
1 parent 07bc404
commit 3856834
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/streamable/iters.py b/streamable/iters.py
@@ -13,6 +13,7 @@
     Coroutine,
     DefaultDict,
     Deque,
+    Generic,
     Iterable,
     Iterator,
     List,
@@ -329,7 +330,7 @@ def __next__(self) -> T:
 
 
 class ConcurrentMappingIterable(
-    ABC, Iterable[Union[U, RaisingIterator.ExceptionContainer]]
+    Generic[T, U], ABC, Iterable[Union[U, RaisingIterator.ExceptionContainer]]
 ):
     """
     Template Method Pattern:
@@ -382,7 +383,7 @@ def __iter__(self) -> Iterator[Union[U, RaisingIterator.ExceptionContainer]]:
                     break
 
 
-class ThreadConcurrentMappingIterable(ConcurrentMappingIterable[U]):
+class ThreadConcurrentMappingIterable(ConcurrentMappingIterable[T, U]):
     def __init__(
         self,
         iterator: Iterator[T],
@@ -418,7 +419,7 @@ def _get_future_result(
         return future.result()
 
 
-class AsyncConcurrentMappingIterable(ConcurrentMappingIterable[U]):
+class AsyncConcurrentMappingIterable(ConcurrentMappingIterable[T, U]):
     def __init__(
         self,
         iterator: Iterator[T],