From 149d9d7a5c63720c681a7edd1fe7161701a1f802 Mon Sep 17 00:00:00 2001
From: Ralf Gabriels <ralfg@hotmail.be>
Date: Tue, 17 Sep 2024 14:30:21 +0200
Subject: [PATCH] Pass verbosity to `predict` and DeepLC Retrainer

---
 deeplc/deeplc.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/deeplc/deeplc.py b/deeplc/deeplc.py
index 57ec84d..bc69007 100644
--- a/deeplc/deeplc.py
+++ b/deeplc/deeplc.py
@@ -610,7 +610,7 @@ def make_preds_core(
         try:
             X
             ret_preds = mod.predict(
-                [X, X_sum, X_global, X_hc], batch_size=self.batch_num_tf
+                [X, X_sum, X_global, X_hc], batch_size=self.batch_num_tf, verbose=int(self.verbose)
             ).flatten()
         except UnboundLocalError:
             logger.debug("X is empty, skipping...")
@@ -1170,6 +1170,7 @@ def calibrate_preds(
                 freeze_layers=True,
                 n_epochs=self.n_epochs,
                 freeze_after_concat=1,
+                verbose=self.verbose,
             )
 
             self.model = models