feat: support Prompt in AccountIdsConfiguration

QuantConnect · Sep 19, 2024 · ed7870f · ed7870f
1 parent d32a8e3
commit ed7870f
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/lean/models/configuration.py b/lean/models/configuration.py
@@ -421,7 +421,7 @@ def ask_user_for_input(self, default_value, logger: Logger, hide_input: bool = F
         raise ValueError(f'user input not allowed with {self.__class__.__name__}')
 
 
-class AccountIdsConfiguration(ChoiceUserInput):
+class AccountIdsConfiguration(PromptUserInput, ChoiceUserInput):
 
     def __init__(self, config_json_object):
         super().__init__(config_json_object)
@@ -447,7 +447,9 @@ def ask_user_for_input(self, default_value, logger: Logger, hide_input: bool = F
         :param hide_input: Whether to hide the input (not used for this type of input, which is never hidden).
         :return: The value provided by the user.
         """
-        if self._input_method == "choice":
+        if self._input_method == "prompt":
+            return PromptUserInput.ask_user_for_input(self, default_value, logger)
+        elif self._input_method == "choice":
             return ChoiceUserInput.ask_user_for_input(self, default_value, logger)
         else:
             raise ValueError(f"Undefined input method type {self._input_method}")