From 89ef31b3811dcb0e53f9f2f643ffb31dd22359f4 Mon Sep 17 00:00:00 2001 From: Robert Caulk Date: Tue, 5 Aug 2025 12:49:27 +0200 Subject: [PATCH] fix: Ensure pipelines that modify feature set are supported for fast backtesting --- freqtrade/freqai/data_drawer.py | 2 +- freqtrade/freqai/freqai_interface.py | 7 +------ 2 files changed, 2 insertions(+), 7 deletions(-) diff --git a/freqtrade/freqai/data_drawer.py b/freqtrade/freqai/data_drawer.py index fef0ab73a..7886bc896 100644 --- a/freqtrade/freqai/data_drawer.py +++ b/freqtrade/freqai/data_drawer.py @@ -493,7 +493,7 @@ class FreqaiDataDrawer: dk.data["data_path"] = str(dk.data_path) dk.data["model_filename"] = str(dk.model_filename) - dk.data["training_features_list"] = list(dk.data_dictionary["train_features"].columns) + dk.data["training_features_list"] = dk.training_features_list dk.data["label_list"] = dk.label_list with (save_path / f"{dk.model_filename}_{METADATA}.json").open("w") as fp: diff --git a/freqtrade/freqai/freqai_interface.py b/freqtrade/freqai/freqai_interface.py index 1c66de46f..b70b7c67e 100644 --- a/freqtrade/freqai/freqai_interface.py +++ b/freqtrade/freqai/freqai_interface.py @@ -514,12 +514,7 @@ class IFreqaiModel(ABC): current coin/bot loop """ - if "training_features_list_raw" in dk.data: - feature_list = dk.data["training_features_list_raw"] - else: - feature_list = dk.data["training_features_list"] - - if dk.training_features_list != feature_list: + if dk.training_features_list != dk.data["training_features_list"]: raise OperationalException( "Trying to access pretrained model with `identifier` " "but found different features furnished by current strategy. "