Update BaseReinforcementLearningModel.py

This commit is contained in:
Robert Caulk 2023-03-08 19:29:39 +01:00 committed by GitHub
parent 29d337fa02
commit 85e345fc48
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -330,8 +330,6 @@ 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
feature_list = [e for e in feature_list if e not in drop_list]
return df