Merge pull request #191 from Yajo/fix-single-prompt-types

fix(types): declare support for single questions
tmbo · Dec 19, 2021 · 1000f1e · 1000f1e
2 parents d13e7e5 + d0b4fa6
commit 1000f1e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/questionary/prompt.py b/questionary/prompt.py
@@ -1,5 +1,5 @@
 from prompt_toolkit.output import ColorDepth
-from typing import Any, Dict, Optional, Iterable, Mapping
+from typing import Any, Dict, Optional, Iterable, Mapping, Union
 
 from questionary import utils
 from questionary.constants import DEFAULT_KBI_MESSAGE
@@ -13,7 +13,7 @@ def __init__(self, message: str, errors: Optional[BaseException] = None) -> None
 
 
 def prompt(
-    questions: Iterable[Mapping[str, Any]],
+    questions: Union[Dict[str, Any], Iterable[Mapping[str, Any]]],
     answers: Optional[Mapping[str, Any]] = None,
     patch_stdout: bool = False,
     true_color: bool = False,
@@ -74,7 +74,7 @@ def prompt(
 
 
 def unsafe_prompt(
-    questions: Iterable[Mapping[str, Any]],
+    questions: Union[Dict[str, Any], Iterable[Mapping[str, Any]]],
     answers: Optional[Mapping[str, Any]] = None,
     patch_stdout: bool = False,
     true_color: bool = False,