Add overloads for get_parser()

brentyi · Oct 14, 2022 · b98b758 · b98b758
1 parent 8634789
commit b98b758
Showing 1 changed file with 22 additions and 0 deletions.
diff --git a/tyro/_cli.py b/tyro/_cli.py
@@ -123,6 +123,28 @@ def cli(
     )
 
 
+@overload
+def get_parser(
+    f: Type[OutT],
+    *,
+    prog: Optional[str] = None,
+    description: Optional[str] = None,
+    default: Optional[OutT] = None,
+) -> argparse.ArgumentParser:
+    ...
+
+
+@overload
+def get_parser(
+    f: Callable[..., OutT],
+    *,
+    prog: Optional[str] = None,
+    description: Optional[str] = None,
+    default: Optional[OutT] = None,
+) -> argparse.ArgumentParser:
+    ...
+
+
 def get_parser(
     f: Union[Type[OutT], Callable[..., OutT]],
     *,