fix pytorch classifier test

2023-03-20 18:04:02 +02:00
parent 9aec1ddb17
commit c00ffcee59
1 changed files with 1 additions and 1 deletions
--- a/tests/freqai/test_freqai_interface.py
+++ b/tests/freqai/test_freqai_interface.py
@@ -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':