From 96b23bf698fa9edf06dc18050a607764ccfe5975 Mon Sep 17 00:00:00 2001 From: Jamie Date: Fri, 30 Apr 2021 15:28:22 -0400 Subject: [PATCH] workaround bt setting mpl backend --- gamestonk_terminal/backtesting/bt_controller.py | 10 +++++++--- gamestonk_terminal/backtesting/bt_helper.py | 1 - 2 files changed, 7 insertions(+), 4 deletions(-) diff --git a/gamestonk_terminal/backtesting/bt_controller.py b/gamestonk_terminal/backtesting/bt_controller.py index 1ebb00e2464..de7f125ca94 100644 --- a/gamestonk_terminal/backtesting/bt_controller.py +++ b/gamestonk_terminal/backtesting/bt_controller.py @@ -4,13 +4,19 @@ __docformat__ = "numpy" import argparse from typing import List, Union from datetime import datetime +import matplotlib as mpl import matplotlib.pyplot as plt from prompt_toolkit.completion import NestedCompleter from gamestonk_terminal import feature_flags as gtff from gamestonk_terminal.helper_funcs import get_flair from gamestonk_terminal.menu import session + +default_backend = mpl.get_backend() +# pylint: disable=wrong-import-position from gamestonk_terminal.backtesting import bt_view +mpl.use(default_backend) + class BacktestingController: """Backtesting Class""" @@ -86,7 +92,7 @@ class BacktestingController: def menu(ticker: str, start: Union[str, datetime]): - """Technical Analysis Menu""" + """Backtesting Menu""" plt.close("all") bt_controller = BacktestingController(ticker, start) bt_controller.call_help(None) @@ -106,9 +112,7 @@ def menu(ticker: str, start: Union[str, datetime]): try: plt.close("all") - process_input = bt_controller.switch(an_input) - if process_input is not None: return process_input diff --git a/gamestonk_terminal/backtesting/bt_helper.py b/gamestonk_terminal/backtesting/bt_helper.py index 6f7dde84fb7..422eb8295ce 100644 --- a/gamestonk_terminal/backtesting/bt_helper.py +++ b/gamestonk_terminal/backtesting/bt_helper.py @@ -4,7 +4,6 @@ __docformat__ = "numpy" from datetime import datetime from typing import Union -import matplotlib from matplotlib import pyplot as plt from pandas.plotting import register_matplotlib_converters import bt