diff --git a/tests/training/test_classification.py b/tests/training/test_classification.py index e28393ea..2d657040 100644 --- a/tests/training/test_classification.py +++ b/tests/training/test_classification.py @@ -507,7 +507,7 @@ def test_lgbm_classification_learner_params(): } ) - lgbm_dataset = lightgbm.Dataset(df[features].values, label=df[target], silent=True) + lgbm_dataset = lightgbm.Dataset(df[features].values, label=df[target]) mock_lgbm = MagicMock() mock_lgbm.predict.return_value = df_result["prediction"]