stream: overload constructor

ebonnal · Jun 7, 2024 · 0474283 · 0474283
1 parent 4768d66
commit 0474283
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/streamable/stream.py b/streamable/stream.py
@@ -37,6 +37,13 @@
 
 
 class Stream(Iterable[T]):
+    # fmt: off
+    @overload
+    def __init__(self, source: Iterable[T]) -> None: ...
+    @overload
+    def __init__(self, source: Callable[[], Iterable[T]]) -> None: ...
+    # fmt: on
+
     def __init__(self, source: Union[Iterable[T], Callable[[], Iterable[T]]]) -> None:
         """
         Initialize a Stream with a data source.