Remove unnecessary models, add model arg

This commit is contained in:
Robert Caulk 2022-09-23 10:37:34 +02:00 committed by GitHub
parent f7dd3045f7
commit 95121550ef
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -177,11 +177,9 @@ def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model):
'LightGBMRegressor', 'LightGBMRegressor',
'XGBoostRegressor', 'XGBoostRegressor',
'CatboostRegressor', 'CatboostRegressor',
'ReinforcementLearner', 'ReinforcementLearner'
'ReinforcementLearner_multiproc',
'ReinforcementLearner_test_4ac'
]) ])
def test_start_backtesting(mocker, freqai_conf): def test_start_backtesting(mocker, freqai_conf, model):
freqai_conf.get("freqai", {}).update({"save_backtest_models": True}) freqai_conf.get("freqai", {}).update({"save_backtest_models": True})
if is_arm() and model == 'CatboostRegressor': if is_arm() and model == 'CatboostRegressor':