fix tests

This commit is contained in:
robcaulk
2022-10-01 20:26:41 +02:00
parent ab9d781b06
commit cf882fa84e
3 changed files with 4 additions and 4 deletions

View File

@@ -213,7 +213,7 @@ class BaseReinforcementLearningModel(IFreqaiModel):
dk.data_dictionary["prediction_features"] = filtered_dataframe
# optional additional data cleaning/analysis
self.data_cleaning_predict(dk, filtered_dataframe)
self.data_cleaning_predict(dk)
pred_df = self.rl_model_predict(
dk.data_dictionary["prediction_features"], dk, self.model)