changed labels

get_forecast.py

@@ -48,11 +48,11 @@ def get_forecast(serie: str, periods, percent_change: int):
     aux_0 = pd.concat([last_obs, values_0])
     
     # arrange dataframe
-    df_future = future_values.rename(columns = {'ds': 'Date', 'yhat': serie}).copy()
+    df_future = future_values.rename(columns = {'ds': 'Date', 'yhat': f'{serie} with price change'}).copy()
     df_future['Date'] = df_future['Date'].apply(lambda x: x.date())
     df_future[f'{serie} without price change'] = values_0.rename(columns = {'ds': 'Date', 'yhat': serie})[serie]
-    df_future['Diff'] = df_future[serie] - df_future[f'{serie} without price change']
-    df_future[serie] = df_future[serie].apply(lambda x: round(x, 4))
+    df_future['Diff'] = df_future[f'{serie} with price change'] - df_future[f'{serie} without price change']
+    df_future[f'{serie} with price change'] = df_future[f'{serie} with price change'].apply(lambda x: round(x, 4))
     df_future[f'{serie} without price change'] = df_future[f'{serie} without price change'].apply(lambda x: round(x, 4))
     df_future['Diff'] = df_future['Diff'].apply(lambda x: round(x, 4))