diff --git a/openbb_terminal/dashboards/voila/forecast.ipynb b/openbb_terminal/dashboards/voila/forecast.ipynb index cc3e2e32936..eb9abd8fee9 100644 --- a/openbb_terminal/dashboards/voila/forecast.ipynb +++ b/openbb_terminal/dashboards/voila/forecast.ipynb @@ -55,6 +55,7 @@ "from IPython.display import display\n", "\n", "from openbb_terminal import api\n", + "openbb = api.openbb\n", "\n", "%matplotlib inline\n", "matplotlib_inline.backend_inline.set_matplotlib_formats(\"svg\")\n", @@ -71,29 +72,31 @@ "try:\n", " has_forecast = True\n", " model_opts = {\n", - " \"expo\": api.forecast.expo,\n", - " \"theta\": api.forecast.theta,\n", - " \"linregr\": api.forecast.linregr,\n", - " \"regr\": api.forecast.regr,\n", - " \"rnn\": api.forecast.rnn,\n", - " \"brnn\": api.forecast.brnn,\n", - " \"nbeats\": api.forecast.nbeats,\n", - " \"tcn\": api.forecast.tcn,\n", - " \"trans\": api.forecast.trans,\n", - " \"tft\": api.forecast.tft,\n", + " \"expo\": openbb.forecast.expo,\n", + " \"theta\": openbb.forecast.theta,\n", + " \"linregr\": openbb.forecast.linregr,\n", + " \"regr\": openbb.forecast.regr,\n", + " \"rnn\": openbb.forecast.rnn,\n", + " \"brnn\": openbb.forecast.brnn,\n", + " \"nbeats\": openbb.forecast.nbeats,\n", + " \"tcn\": openbb.forecast.tcn,\n", + " \"trans\": openbb.forecast.trans,\n", + " \"tft\": openbb.forecast.tft,\n", + " \"nhits\": openbb.forecast.nhits\n", " }\n", "\n", " feat_engs = {\n", - " \"ema\": api.forecast.ema,\n", - " \"sto\": api.forecast.sto,\n", - " \"rsi\": api.forecast.rsi,\n", - " \"roc\": api.forecast.roc,\n", - " \"mom\": api.forecast.mom,\n", - " \"atr\": api.forecast.atr,\n", - " \"delta\": api.forecast.delta,\n", - " \"signal\": api.forecast.signal,\n", + " \"ema\": openbb.forecast.ema,\n", + " \"sto\": openbb.forecast.sto,\n", + " \"rsi\": openbb.forecast.rsi,\n", + " \"roc\": openbb.forecast.roc,\n", + " \"mom\": openbb.forecast.mom,\n", + " \"atr\": openbb.forecast.atr,\n", + " \"delta\": openbb.forecast.delta,\n", + " \"signal\": openbb.forecast.signal,\n", " }\n", - "except AttributeError:\n", + "except AttributeError as e:\n", + " print(e)\n", " has_forecast = False\n", " model_opts = {}\n", " feat_engs = {}\n", @@ -207,7 +210,7 @@ " result = calcs\n", " if not target_column:\n", " target_column = self.df.columns[0]\n", - " kwargs = {}\n", + " kwargs = {\"chart\": True}\n", " if contains_covariates and past_covariates != ():\n", " kwargs[\"past_covariates\"] = \",\".join(past_covariates)\n", " if has_parameter(forecast_model, \"naive\"):\n", diff --git a/openbb_terminal/dashboards/voila/vsurf.ipynb b/openbb_terminal/dashboards/voila/vsurf.ipynb index 816c8b52128..906c382528b 100644 --- a/openbb_terminal/dashboards/voila/vsurf.ipynb +++ b/openbb_terminal/dashboards/voila/vsurf.ipynb @@ -22,7 +22,7 @@ "if module_path not in sys.path:\n", " sys.path.append(module_path)\n", "\n", - "from openbb_terminal import api as openbb" + "from openbb_terminal.api import openbb" ] }, { @@ -73,7 +73,7 @@ " def create(self, z, ticker, height, width):\n", " if ticker:\n", " if ticker != self.last_ticker:\n", - " self.df = openbb.stocks.options.models.yfinance.get_iv_surface(ticker)\n", + " self.df = openbb.stocks.options.vsurf(ticker)\n", " self.last_ticker = ticker\n", "\n", " if not self.df.empty:\n", @@ -154,7 +154,7 @@ "name": "python", "nbconvert_exporter": "python", "pygments_lexer": "ipython3", - "version": "3.8.13" + "version": "3.9.6" } }, "nbformat": 4,