From eba82360fac3776bcaed85e44d66e58292fa02df Mon Sep 17 00:00:00 2001 From: Yinon Polak Date: Tue, 21 Mar 2023 15:18:05 +0200 Subject: [PATCH] skip pytorch tests on python 3.11 and intel based mac os --- tests/freqai/test_freqai_interface.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index 718d80f44..b4d808af2 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -146,8 +146,7 @@ def test_extract_data_and_train_model_Standard(mocker, freqai_conf, model, pca, ('CatboostClassifierMultiTarget', "freqai_test_multimodel_classifier_strat") ]) def test_extract_data_and_train_model_MultiTargets(mocker, freqai_conf, model, strat): - if (is_arm() or is_py11()) and 'Catboost' in model: - pytest.skip("CatBoost is not supported on ARM") + can_run_model(model) freqai_conf.update({"timerange": "20180110-20180130"}) freqai_conf.update({"strategy": strat}) @@ -189,8 +188,7 @@ def test_extract_data_and_train_model_MultiTargets(mocker, freqai_conf, model, s 'PyTorchMLPClassifier', ]) def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model): - if (is_arm() or is_py11()) and model == 'CatboostClassifier': - pytest.skip("CatBoost is not supported on ARM") + can_run_model(model) freqai_conf.update({"freqaimodel": model}) freqai_conf.update({"strategy": "freqai_test_classifier"})