Skip to content

Commit 51f3f0d

Browse files
committed
test: Remove catboost tests
1 parent 006d230 commit 51f3f0d

File tree

1 file changed

+0
-9
lines changed

1 file changed

+0
-9
lines changed

tests/freqai/test_freqai_interface.py

Lines changed: 0 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -31,9 +31,6 @@
3131
def can_run_model(model: str) -> None:
3232
is_pytorch_model = "Reinforcement" in model or "PyTorch" in model
3333

34-
if is_arm() and "Catboost" in model:
35-
pytest.skip("CatBoost is not supported on ARM.")
36-
3734
if is_pytorch_model and is_mac():
3835
pytest.skip("Reinforcement learning / PyTorch module not available on intel based Mac OS.")
3936

@@ -44,7 +41,6 @@ def can_run_model(model: str) -> None:
4441
("LightGBMRegressor", True, False, True, True, False, 0, 0),
4542
("XGBoostRegressor", False, True, False, True, False, 10, 0.05),
4643
("XGBoostRFRegressor", False, False, False, True, False, 0, 0),
47-
("CatboostRegressor", False, False, False, True, True, 0, 0),
4844
("PyTorchMLPRegressor", False, False, False, False, False, 0, 0),
4945
("PyTorchTransformerRegressor", False, False, False, False, False, 0, 0),
5046
("ReinforcementLearner", False, True, False, True, False, 0, 0),
@@ -138,9 +134,7 @@ def test_extract_data_and_train_model_Standard(
138134
[
139135
("LightGBMRegressorMultiTarget", "freqai_test_multimodel_strat"),
140136
("XGBoostRegressorMultiTarget", "freqai_test_multimodel_strat"),
141-
("CatboostRegressorMultiTarget", "freqai_test_multimodel_strat"),
142137
("LightGBMClassifierMultiTarget", "freqai_test_multimodel_classifier_strat"),
143-
("CatboostClassifierMultiTarget", "freqai_test_multimodel_classifier_strat"),
144138
],
145139
)
146140
@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
184178
"model",
185179
[
186180
"LightGBMClassifier",
187-
"CatboostClassifier",
188181
"XGBoostClassifier",
189182
"XGBoostRFClassifier",
190183
"SKLearnRandomForestClassifier",
@@ -246,13 +239,11 @@ def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model):
246239
[
247240
("LightGBMRegressor", 2, "freqai_test_strat"),
248241
("XGBoostRegressor", 2, "freqai_test_strat"),
249-
("CatboostRegressor", 2, "freqai_test_strat"),
250242
("PyTorchMLPRegressor", 2, "freqai_test_strat"),
251243
("PyTorchTransformerRegressor", 2, "freqai_test_strat"),
252244
("ReinforcementLearner", 3, "freqai_rl_test_strat"),
253245
("XGBoostClassifier", 2, "freqai_test_classifier"),
254246
("LightGBMClassifier", 2, "freqai_test_classifier"),
255-
("CatboostClassifier", 2, "freqai_test_classifier"),
256247
("PyTorchMLPClassifier", 2, "freqai_test_classifier"),
257248
],
258249
)

0 commit comments

Comments
 (0)