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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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