bug fix

app.py

@@ -62,7 +62,7 @@ with open('data/parameters.pkl', 'rb') as f:
 model = TemporalFusionTransformer.load_from_checkpoint('model/tft_check.ckpt', map_location=torch.device('cpu'))
 
 df = pd.read_pickle('data/test_data.pkl')
-df = df.loc[(df["Branch"] == 15) & (df["Group"].isin(["6","7","4","1"]))]
+df = df.loc[(df["Branch"] == "15") & (df["Group"].isin(["6","7","4","1"]))]
 
 rain_mapping = {
     "Yes" : 1,
@@ -121,10 +121,10 @@ st.pyplot(fig)
 st.button("Forecast Sales", type="primary") #on_click=None,
 
 # %%
-preds = raw_preds_to_df(out, quantiles = None)
+#preds = raw_preds_to_df(out, quantiles = None)
 
-preds = preds.merge(data_selected[['time_idx','Group','Branch','sales','weight','Date','MTXWTH_Day_precip','MTXWTH_Temp_max','MTXWTH_Temp_min']],how='left',left_on=['pred_idx','Group','Branch'],right_on=['time_idx','Group','Branch'])
-preds.rename(columns={'time_idx_x':'time_idx'},inplace=True)
-preds.drop(columns=['time_idx_y'],inplace=True)
+#preds = preds.merge(data_selected[['time_idx','Group','Branch','sales','weight','Date','MTXWTH_Day_precip','MTXWTH_Temp_max','MTXWTH_Temp_min']],how='left',left_on=['pred_idx','Group','Branch'],right_on=['time_idx','Group','Branch'])
+#preds.rename(columns={'time_idx_x':'time_idx'},inplace=True)
+#preds.drop(columns=['time_idx_y'],inplace=True)