[FIX] Silence catboost (#338)

* set verbose=False in catboost * fix flake
automl · Nov 22, 2021 · 751b9ce · 751b9ce
1 parent 10df650
commit 751b9ce
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py b/autoPyTorch/pipeline/components/setup/traditional_ml/traditional_learner/learners.py
@@ -135,7 +135,11 @@ def _fit(self, X_train: np.ndarray,
         X_train_pooled = Pool(data=X_train, label=y_train, cat_features=categoricals)
         X_val_pooled = Pool(data=X_val, label=y_val, cat_features=categoricals)
 
-        self.model.fit(X_train_pooled, eval_set=X_val_pooled, use_best_model=True, early_stopping_rounds=early_stopping)
+        self.model.fit(X_train_pooled,
+                       eval_set=X_val_pooled,
+                       use_best_model=True,
+                       early_stopping_rounds=early_stopping,
+                       verbose=False)
 
     @staticmethod
     def get_properties(