fix: pandas.

This commit is contained in:
robcaulk
2024-04-07 14:20:52 +02:00
parent 08d0e76187
commit d8aaaa34d3

View File

@@ -285,7 +285,7 @@ class FreqaiDataDrawer:
new_pred["date_pred"] = dataframe["date"]
# set everything to nan except date_pred
columns_to_nan = new_pred.columns.difference(['date_pred'])
columns_to_nan = new_pred.columns.difference(['date_pred', 'date'])
new_pred[columns_to_nan] = new_pred[columns_to_nan].astype(
float).values * np.nan