From 8e18d4729820452d3d866bc8098520efa28aaaad Mon Sep 17 00:00:00 2001 From: James Simmons <53658028+simmonsj330@users.noreply.github.com> Date: Wed, 16 Nov 2022 10:56:35 -0500 Subject: [PATCH] Fixing `exit` command on forecast menu (#3421) * adding additional exit to queue * changing forecast controller Co-authored-by: James Maslek --- openbb_terminal/forecast/autoselect_model.py | 2 +- openbb_terminal/forecast/forecast_controller.py | 6 ++++++ 2 files changed, 7 insertions(+), 1 deletion(-) diff --git a/openbb_terminal/forecast/autoselect_model.py b/openbb_terminal/forecast/autoselect_model.py index 3ad86e5f176..40387e19483 100644 --- a/openbb_terminal/forecast/autoselect_model.py +++ b/openbb_terminal/forecast/autoselect_model.py @@ -19,7 +19,7 @@ warnings.simplefilter("ignore") logger = logging.getLogger(__name__) -# pylint: disable=E1123 +# pylint: disable=E1123,E1137 def precision_format(best_model: str, index: str, val: float) -> str: diff --git a/openbb_terminal/forecast/forecast_controller.py b/openbb_terminal/forecast/forecast_controller.py index 99f631404b6..a0ba05b6487 100644 --- a/openbb_terminal/forecast/forecast_controller.py +++ b/openbb_terminal/forecast/forecast_controller.py @@ -227,6 +227,12 @@ class ForecastController(BaseController): self.completer = NestedCompleter.from_nested_dict(choices) self.update_runtime_choices() + def call_exit(self, _) -> None: + """Process exit terminal command from forecast menu.""" + self.save_class() + for _ in range(self.PATH.count("/") + 1): + self.queue.insert(0, "quit") + def update_runtime_choices(self): # Load in any newly exported files self.DATA_FILES = forecast_model.get_default_files()