From 81a2cbb4eb65c7b2a7df3b3da93a0b5d7c87800a Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Mon, 20 Mar 2023 18:10:17 +0200 Subject: [PATCH] fix tests --- tests/freqai/test_freqai_interface.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index d35b00013..7931dc7a4 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -48,7 +48,7 @@ def can_run_model(model: str) -> None: ('XGBoostRegressor', False, True, False, True, False, 10), ('XGBoostRFRegressor', False, False, False, True, False, 0), ('CatboostRegressor', False, False, False, True, True, 0), - ('MLPPyTorchRegressor', False, False, False, True, False, 0), + ('PyTorchMLPRegressor', False, False, False, True, False, 0), ('ReinforcementLearner', False, True, False, True, False, 0), ('ReinforcementLearner_multiproc', False, False, False, True, False, 0), ('ReinforcementLearner_test_3ac', False, False, False, False, False, 0), @@ -86,7 +86,7 @@ def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model, pca, if 'test_3ac' in model or 'test_4ac' in model: freqai_conf["freqaimodel_path"] = str(Path(__file__).parents[1] / "freqai" / "test_models") - if 'MLPPyTorchRegressor' in model: + if 'PyTorchMLPRegressor' in model: model_save_ext = 'zip' freqai_conf['freqai']['model_training_parameters'].update({ "learning_rate": 3e-4, @@ -214,7 +214,7 @@ def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model): freqai.extract_data_and_train_model(new_timerange, "ADA/BTC", strategy, freqai.dk, data_load_timerange) - if 'MLPPyTorchClassifier': + if 'PyTorchMLPClassifier': freqai_conf['freqai']['model_training_parameters'].update({ "learning_rate": 3e-4, "trainer_kwargs": {