Fix tapify.py

swansonk14 · Oct 18, 2023 · 612fabe · 612fabe
1 parent 18b13d1
commit 612fabe
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tap/tapify.py b/tap/tapify.py
@@ -1,6 +1,6 @@
 """Tapify module, which can initialize a class or run a function by parsing arguments from the command line."""
 from inspect import signature, Parameter
-from typing import Any, Callable, List, Optional, TypeVar, Union
+from typing import Any, Callable, List, Optional, Type, TypeVar, Union
 
 from docstring_parser import parse
 
@@ -10,7 +10,7 @@
 OutputType = TypeVar('OutputType')
 
 
-def tapify(class_or_function: Union[Callable[[InputType], OutputType], OutputType],
+def tapify(class_or_function: Union[Callable[[InputType], OutputType], Type[OutputType]],
            known_only: bool = False,
            command_line_args: Optional[List[str]] = None,
            **func_kwargs) -> OutputType: