Fix classification tests

nubank · Nov 7, 2023 · 1961853 · 1961853
1 parent 15183e5
commit 1961853
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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"]