Fix for progress bar crash with lightning 2.2

Fixes one part of #592
mittagessen · Apr 22, 2024 · aca1779 · aca1779
1 parent 3ed6526
commit aca1779
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/kraken/lib/progress.py b/kraken/lib/progress.py
@@ -110,7 +110,10 @@ def _init_progress(self, trainer):
             reconfigure(**self._console_kwargs)
             self._console = get_console()
             self._console.clear_live()
-            self._metric_component = MetricsTextColumn(trainer, self.theme.metrics)
+            self._metric_component = MetricsTextColumn(trainer,
+                                                       self.theme.metrics,
+                                                       self.theme.metrics_text_delimiter,
+                                                       self.theme.metrics_format)
             columns = self.configure_columns(trainer)
             columns.append(self._metric_component)
 
@@ -158,3 +161,5 @@ class RichProgressBarTheme:
     time: Union[str, 'Style'] = DEFAULT_STYLES['progress.elapsed']
     processing_speed: Union[str, 'Style'] = DEFAULT_STYLES['progress.data.speed']
     metrics: Union[str, 'Style'] = DEFAULT_STYLES['progress.description']
+    metrics_text_delimiter: str = ' '
+    metrics_format: str = '.3f'