fix pytorch classifier test

This commit is contained in:
Yinon Polak 2023-03-20 18:04:02 +02:00
parent 9aec1ddb17
commit c00ffcee59

View File

@ -168,7 +168,7 @@ def test_extract_data_and_train_model_MultiTargets(mocker, freqai_conf, model, s
'CatboostClassifier', 'CatboostClassifier',
'XGBoostClassifier', 'XGBoostClassifier',
'XGBoostRFClassifier', 'XGBoostRFClassifier',
'MLPPyTorchClassifier', 'PyTorchMLPClassifier',
]) ])
def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model): def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model):
if (is_arm() or is_py11()) and model == 'CatboostClassifier': if (is_arm() or is_py11()) and model == 'CatboostClassifier':