fix: Ensure pipelines that modify feature set are supported for fast backtesting

This commit is contained in:
Robert Caulk
2025-08-05 12:49:27 +02:00
parent d1c901ae94
commit 89ef31b381
2 changed files with 2 additions and 7 deletions

View File

@@ -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:

View File

@@ -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. "