diff --git a/skops/io/tests/test_external.py b/skops/io/tests/test_external.py index fc58f5f4..f9edcffc 100644 --- a/skops/io/tests/test_external.py +++ b/skops/io/tests/test_external.py @@ -282,7 +282,9 @@ def trusted(self): @pytest.mark.parametrize("boosting_type", boosting_types) def test_classifier(self, catboost, cb_clf_data, trusted, boosting_type): - estimator = catboost.CatBoostClassifier(boosting_type=boosting_type) + estimator = catboost.CatBoostClassifier( + verbose=False, boosting_type=boosting_type + ) loaded = loads(dumps(estimator), trusted=trusted) assert_params_equal(estimator.get_params(), loaded.get_params()) @@ -293,7 +295,9 @@ def test_classifier(self, catboost, cb_clf_data, trusted, boosting_type): @pytest.mark.parametrize("boosting_type", boosting_types) def test_regressor(self, catboost, cb_regr_data, trusted, boosting_type): - estimator = catboost.CatBoostRegressor(boosting_type=boosting_type) + estimator = catboost.CatBoostRegressor( + verbose=False, boosting_type=boosting_type + ) loaded = loads(dumps(estimator), trusted=trusted) assert_params_equal(estimator.get_params(), loaded.get_params()) @@ -304,7 +308,7 @@ def test_regressor(self, catboost, cb_regr_data, trusted, boosting_type): @pytest.mark.parametrize("boosting_type", boosting_types) def test_ranker(self, catboost, cb_rank_data, trusted, boosting_type): - estimator = catboost.CatBoostRanker(boosting_type=boosting_type) + estimator = catboost.CatBoostRanker(verbose=False, boosting_type=boosting_type) loaded = loads(dumps(estimator), trusted=trusted) assert_params_equal(estimator.get_params(), loaded.get_params())