From bfad63e6934e8039b0f8dec2be60e1664d355768 Mon Sep 17 00:00:00 2001 From: Aaron Cutright Date: Mon, 15 Mar 2021 20:51:14 -0400 Subject: [PATCH] Fix uncommon column case issue for fbprophet 'date' was sometimes 'Date' --- gamestonk_terminal/prediction_techniques/fbprophet.py | 1 + 1 file changed, 1 insertion(+) diff --git a/gamestonk_terminal/prediction_techniques/fbprophet.py b/gamestonk_terminal/prediction_techniques/fbprophet.py index 21447c7d1ca..ae4d9d0218f 100644 --- a/gamestonk_terminal/prediction_techniques/fbprophet.py +++ b/gamestonk_terminal/prediction_techniques/fbprophet.py @@ -95,6 +95,7 @@ def fbprophet(l_args, s_ticker, df_stock): df_stock = df_stock.sort_index(ascending=True) df_stock.reset_index(level=0, inplace=True) + df_stock.columns = map(str.lower, df_stock.columns) # column names are sometimes upper cased df_stock = df_stock[["date", "5. adjusted close"]] df_stock = df_stock.rename(columns={"date": "ds", "5. adjusted close": "y"}) df_stock["ds"] = pd.to_datetime(df_stock["ds"])