Remove unnecessary models, add model arg

2022-09-23 10:37:34 +02:00
parent f7dd3045f7
commit 95121550ef
1 changed files with 2 additions and 4 deletions
--- 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':