3131def 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