From 95121550efd6417583f6bfa2ab251c564e22d5d8 Mon Sep 17 00:00:00 2001 From: Robert Caulk Date: Fri, 23 Sep 2022 10:37:34 +0200 Subject: [PATCH] Remove unnecessary models, add model arg --- tests/freqai/test_freqai_interface.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index 592499a34..ac8fd2b42 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -177,11 +177,9 @@ def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model): 'LightGBMRegressor', 'XGBoostRegressor', 'CatboostRegressor', - 'ReinforcementLearner', - 'ReinforcementLearner_multiproc', - 'ReinforcementLearner_test_4ac' + 'ReinforcementLearner' ]) -def test_start_backtesting(mocker, freqai_conf): +def test_start_backtesting(mocker, freqai_conf, model): freqai_conf.get("freqai", {}).update({"save_backtest_models": True}) if is_arm() and model == 'CatboostRegressor':