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',
'XGBoostClassifier',
'XGBoostRFClassifier',
'MLPPyTorchClassifier',
'PyTorchMLPClassifier',
])
def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model):
if (is_arm() or is_py11()) and model == 'CatboostClassifier':