Merge pull request #214 from ionite34/type-hints

Add return type annotations to API #214
pavdmyt · Oct 11, 2022 · 77ba3e3 · 77ba3e3
2 parents 5692862 + 8ae7b2c
commit 77ba3e3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/yaspin/api.py b/yaspin/api.py
@@ -14,7 +14,7 @@
 from .signal_handlers import default_handler
 
 
-def yaspin(*args, **kwargs):
+def yaspin(*args, **kwargs) -> Yaspin:
     """Display spinner in stdout.
 
     Can be used as a context manager or as a function decorator.
@@ -78,7 +78,7 @@ def foo():
     return Yaspin(*args, **kwargs)
 
 
-def kbi_safe_yaspin(*args, **kwargs):
+def kbi_safe_yaspin(*args, **kwargs) -> Yaspin:
     kwargs["sigmap"] = {signal.SIGINT: default_handler}
     return Yaspin(*args, **kwargs)