Spaces:

leuschnm
/

TemporalFusionTransformer

Runtime error

leuschnm commited on Jul 2, 2023

Commit

a3c8610

1 Parent(s): b174ce6

bug fix

Files changed (1) hide show

app.py CHANGED Viewed

@@ -62,7 +62,7 @@ def predict(_model, _dataloader, datepicker):
         group["date_imputed"] = [datepicker + datetime.timedelta(days=x) for x in range(30)]
         return group
-    preds["date_imputed"] = preds.groupby("Group").transform(add_dates)
     return preds[["date_imputed", "Group", "pred"]]

         group["date_imputed"] = [datepicker + datetime.timedelta(days=x) for x in range(30)]
         return group
+    preds["date_imputed"] = preds.groupby("Group").pred.transform(add_dates)
     return preds[["date_imputed", "Group", "pred"]]