diff --git a/moe/util/cli/prompt.py b/moe/util/cli/prompt.py index 5ea4e658..319393b9 100644 --- a/moe/util/cli/prompt.py +++ b/moe/util/cli/prompt.py @@ -62,7 +62,9 @@ def choice_prompt( ) ) - user_input = questionary.rawselect(question, choices=questionary_choices).ask() + user_input = questionary.select( + question, choices=questionary_choices, use_shortcuts=True, use_arrow_keys=True + ).ask() for prompt_choice in prompt_choices: if prompt_choice.shortcut_key == user_input: diff --git a/tests/util/cli/test_prompt.py b/tests/util/cli/test_prompt.py index 3f05d337..fffae36d 100644 --- a/tests/util/cli/test_prompt.py +++ b/tests/util/cli/test_prompt.py @@ -20,7 +20,7 @@ def test_prompt_choice_return(self): assert track.title != "a" with patch( - "moe.util.cli.prompt.questionary.rawselect", + "moe.util.cli.prompt.questionary.select", **{"return_value.ask.return_value": "a"} ): prompt_choice = choice_prompt([mock_choice1, mock_choice2]) @@ -33,7 +33,7 @@ def test_invalid_input(self): mock_choice = PromptChoice("title b", "b", lambda b: None) with patch( - "moe.util.cli.prompt.questionary.rawselect", + "moe.util.cli.prompt.questionary.select", **{"return_value.ask.return_value": "a"} ): with pytest.raises(SystemExit) as error: