ensure training_features_list is updated properly

2023-03-08 19:37:11 +01:00 · 2023-03-08 19:37:11 +01:00 · d10ee0979a
commit d10ee0979a
parent 85e345fc48
1 changed files with 2 additions and 0 deletions
--- a/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
+++ b/freqtrade/freqai/RL/BaseReinforcementLearningModel.py
@ -330,6 +330,8 @@ class BaseReinforcementLearningModel(IFreqaiModel):

        if self.rl_config["drop_ohlc_from_features"]:
            df.drop(drop_list, axis=1, inplace=True)
+            feature_list = dk.training_features_list
+            dk.training_features_list = [e for e in feature_list if e not in drop_list]

        return df