Update data_drawer.py

This commit is contained in:
Robert Caulk
2023-12-05 12:16:08 +01:00
committed by GitHub
parent 8b5e6bf9be
commit caac77c90b

View File

@@ -285,8 +285,8 @@ class FreqaiDataDrawer:
hist_preds = self.historic_predictions[pair].copy()
# ensure both dataframes have the same date format so they can be merged
new_pred["date_pred"] = pd.to_datetime(new_pred["date_pred"], utc=True)
hist_preds["date_pred"] = pd.to_datetime(hist_preds["date_pred"], utc=True)
new_pred["date_pred"] = pd.to_datetime(new_pred["date_pred"])
hist_preds["date_pred"] = pd.to_datetime(hist_preds["date_pred"])
# find the closest common date between new_pred and historic predictions
# and cut off the new_pred dataframe at that date