From e94c1f894dd216a705e647c4eb4579e8b7a5a52b Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Eren=20G=C3=B6lge?= <egolge@coqui.ai>
Date: Sat, 7 Aug 2021 21:54:19 +0000
Subject: [PATCH] Simplify `console_logger`

---
 TTS/utils/logging/console_logger.py | 9 +--------
 1 file changed, 1 insertion(+), 8 deletions(-)

diff --git a/TTS/utils/logging/console_logger.py b/TTS/utils/logging/console_logger.py
index bb6644c9..c5fbe8b4 100644
--- a/TTS/utils/logging/console_logger.py
+++ b/TTS/utils/logging/console_logger.py
@@ -38,7 +38,7 @@ class ConsoleLogger:
     def print_train_start(self):
         print(f"\n{tcolors.BOLD} > TRAINING ({self.get_time()}) {tcolors.ENDC}")
 
-    def print_train_step(self, batch_steps, step, global_step, log_dict, loss_dict, avg_loss_dict):
+    def print_train_step(self, batch_steps, step, global_step, loss_dict, avg_loss_dict):
         indent = "     | > "
         print()
         log_text = "{}   --> STEP: {}/{} -- GLOBAL_STEP: {}{}\n".format(
@@ -50,13 +50,6 @@ class ConsoleLogger:
                 log_text += "{}{}: {:.5f}  ({:.5f})\n".format(indent, key, value, avg_loss_dict[f"avg_{key}"])
             else:
                 log_text += "{}{}: {:.5f} \n".format(indent, key, value)
-        for idx, (key, value) in enumerate(log_dict.items()):
-            if isinstance(value, list):
-                log_text += f"{indent}{key}: {value[0]:.{value[1]}f}"
-            else:
-                log_text += f"{indent}{key}: {value}"
-            if idx < len(log_dict) - 1:
-                log_text += "\n"
         print(log_text, flush=True)
 
     # pylint: disable=unused-argument