From d8aaaa34d3f18b9c9d12468d69911f6318225d8e Mon Sep 17 00:00:00 2001 From: robcaulk Date: Sun, 7 Apr 2024 14:20:52 +0200 Subject: [PATCH] fix: pandas. --- freqtrade/freqai/data_drawer.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/freqtrade/freqai/data_drawer.py b/freqtrade/freqai/data_drawer.py index 50cc70fe0..4df52fdc4 100644 --- a/freqtrade/freqai/data_drawer.py +++ b/freqtrade/freqai/data_drawer.py @@ -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