.skip/.distinct: fix genericity of return type

ebonnal · Jan 16, 2025 · 7f9a9a0 · 7f9a9a0
1 parent b4b9cc0
commit 7f9a9a0
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/streamable/stream.py b/streamable/stream.py
@@ -165,7 +165,7 @@ def display(self, level: int = logging.INFO) -> "Stream[T]":
 
     def distinct(
         self, key: Optional[Callable[[T], Any]] = None, consecutive_only: bool = False
-    ) -> "Stream":
+    ) -> "Stream[T]":
         """
         Filters the stream to yield only distinct elements.
         If a deduplication `key` is specified, `foo` and `bar` are treated as duplicates when `key(foo) == key(bar)`.
@@ -429,7 +429,7 @@ def observe(self, what: str = "elements") -> "Stream[T]":
         """
         return ObserveStream(self, what)
 
-    def skip(self, count: int) -> "Stream":
+    def skip(self, count: int) -> "Stream[T]":
         """
         Skips the first `count` elements.