🧑‍💻 Fix autocomplete not working missing typehint

Add type hint in wrapper functions for BaseStream. Fixes not working autocomplete.
PickwickSoft · Aug 29, 2023 · bea7f31 · bea7f31
1 parent 682f5b1
commit bea7f31
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()