From c00ffcee59e60f3218b5e09258ad15d57a439ead Mon Sep 17 00:00:00 2001
From: Yinon Polak <inon51@gmail.com>
Date: Mon, 20 Mar 2023 18:04:02 +0200
Subject: [PATCH] fix pytorch classifier test

---
 tests/freqai/test_freqai_interface.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py
index d183501ea..01aa0d1db 100644
--- 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':