Spaces:

leuschnm
/

TemporalFusionTransformer

Runtime error

leuschnm commited on Jul 3, 2023

Commit

d23039a

1 Parent(s): 4c1e191

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -51,7 +51,7 @@ def prepare_dataset(_parameters, df, rain, temperature, datepicker, mapping):
     df = df.loc[(df["Date"].dt.date>lowerbound) & (df["Date"].dt.date<=upperbound)]
-    df = TimeSeriesDataSet.from_parameters(parameters, df)
     return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
 @st.cache_data

     df = df.loc[(df["Date"].dt.date>lowerbound) & (df["Date"].dt.date<=upperbound)]
+    df = TimeSeriesDataSet.from_parameters(_parameters, df)
     return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
 @st.cache_data