Merge pull request #68 from PickwickSoft/bugfix/fix-autocomplete-type…

…hints-decorator 🧑‍💻 Fix autocomplete not working missing typehint
PickwickSoft · Aug 29, 2023 · 8340e5f · 8340e5f
2 parents 682f5b1 + bea7f31
commit 8340e5f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pystreamapi/_streams/__base_stream.py b/pystreamapi/_streams/__base_stream.py
@@ -30,7 +30,7 @@ def _operation(func):
     To be applied to intermediate operations.
     """
     @functools.wraps(func)
-    def wrapper(*args, **kwargs):
+    def wrapper(*args, **kwargs) -> BaseStream[K]:
         self: BaseStream = args[0]
         self._verify_open()
         return func(*args, **kwargs)
@@ -45,7 +45,7 @@ def terminal(func):
     """
     @functools.wraps(func)
     @_operation
-    def wrapper(*args, **kwargs):
+    def wrapper(*args, **kwargs) -> BaseStream[K]:
         self: BaseStream = args[0]
         self._queue.execute_all()
         self._close()