From 51f3f0d65ad1d4e95e818ff73f40440564452681 Mon Sep 17 00:00:00 2001 From: Matthias Date: Mon, 22 Dec 2025 19:53:02 +0100 Subject: [PATCH] test: Remove catboost tests --- tests/freqai/test_freqai_interface.py | 9 --------- 1 file changed, 9 deletions(-) diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index 3e7e96d85..053554dcc 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -31,9 +31,6 @@ from tests.freqai.conftest import ( def can_run_model(model: str) -> None: is_pytorch_model = "Reinforcement" in model or "PyTorch" in model - if is_arm() and "Catboost" in model: - pytest.skip("CatBoost is not supported on ARM.") - if is_pytorch_model and is_mac(): pytest.skip("Reinforcement learning / PyTorch module not available on intel based Mac OS.") @@ -44,7 +41,6 @@ def can_run_model(model: str) -> None: ("LightGBMRegressor", True, False, True, True, False, 0, 0), ("XGBoostRegressor", False, True, False, True, False, 10, 0.05), ("XGBoostRFRegressor", False, False, False, True, False, 0, 0), - ("CatboostRegressor", False, False, False, True, True, 0, 0), ("PyTorchMLPRegressor", False, False, False, False, False, 0, 0), ("PyTorchTransformerRegressor", False, False, False, False, False, 0, 0), ("ReinforcementLearner", False, True, False, True, False, 0, 0), @@ -138,9 +134,7 @@ def test_extract_data_and_train_model_Standard( [ ("LightGBMRegressorMultiTarget", "freqai_test_multimodel_strat"), ("XGBoostRegressorMultiTarget", "freqai_test_multimodel_strat"), - ("CatboostRegressorMultiTarget", "freqai_test_multimodel_strat"), ("LightGBMClassifierMultiTarget", "freqai_test_multimodel_classifier_strat"), - ("CatboostClassifierMultiTarget", "freqai_test_multimodel_classifier_strat"), ], ) @pytest.mark.filterwarnings(r"ignore:.*__sklearn_tags__.*:DeprecationWarning") @@ -184,7 +178,6 @@ def test_extract_data_and_train_model_MultiTargets(mocker, freqai_conf, model, s "model", [ "LightGBMClassifier", - "CatboostClassifier", "XGBoostClassifier", "XGBoostRFClassifier", "SKLearnRandomForestClassifier", @@ -246,13 +239,11 @@ def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model): [ ("LightGBMRegressor", 2, "freqai_test_strat"), ("XGBoostRegressor", 2, "freqai_test_strat"), - ("CatboostRegressor", 2, "freqai_test_strat"), ("PyTorchMLPRegressor", 2, "freqai_test_strat"), ("PyTorchTransformerRegressor", 2, "freqai_test_strat"), ("ReinforcementLearner", 3, "freqai_rl_test_strat"), ("XGBoostClassifier", 2, "freqai_test_classifier"), ("LightGBMClassifier", 2, "freqai_test_classifier"), - ("CatboostClassifier", 2, "freqai_test_classifier"), ("PyTorchMLPClassifier", 2, "freqai_test_classifier"), ], )