printing alongside progress bar added to LightningModule.print()

Author: Alexander Snorkin

pytorch_lightning/core/lightning.py

@@ -170,7 +170,11 @@ def forward(self, x):
 
         """
         if self.trainer.is_global_zero:
-            print(*args, **kwargs)
+            progress_bar = self.trainer.progress_bar_callback
+            if progress_bar is not None and progress_bar.is_enabled:
+                progress_bar.print(*args, **kwargs)
+            else:
+                print(*args, **kwargs)
 
     def log(
         self,