leuschnm commited on
Commit
bb5ae00
·
1 Parent(s): 89c3cca
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -48,7 +48,7 @@ def prepare_dataset(parameters, df, rain, temperature, datepicker):
48
  lowerbound = datepicker - datetime.timedelta(days = 35)
49
  upperbound = datepicker + datetime.timedelta(days = 30)
50
 
51
- df = df.loc[(df["Date"]>lowerbound) & (df["Date"]<=upperbound)]
52
 
53
  df = TimeSeriesDataSet.from_parameters(parameters, df)
54
  return df.to_dataloader(train=False, batch_size=256,num_workers = 0)
 
48
  lowerbound = datepicker - datetime.timedelta(days = 35)
49
  upperbound = datepicker + datetime.timedelta(days = 30)
50
 
51
+ df = df.loc[(df["Date"].dt.date>lowerbound) & (df["Date"].dt.date<=upperbound)]
52
 
53
  df = TimeSeriesDataSet.from_parameters(parameters, df)
54
  return df.to_dataloader(train=False, batch_size=256,num_workers = 0)