From 23622d765db16d9292bf03491d6d262d8c910980 Mon Sep 17 00:00:00 2001 From: Paul Wagener Date: Sun, 1 Sep 2024 22:33:25 +0200 Subject: [PATCH] Don't panic when the progress is > 1.0 (#2229) Ratatui asserts that gauges don't have a progress greater than 1.0 This can happen if a dataset reports a lower len() than it actually provides. This change prevents a panic when the `Progress::items_processed` is greater than the `Progress::items_total` --- crates/burn-train/src/renderer/tui/progress.rs | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/crates/burn-train/src/renderer/tui/progress.rs b/crates/burn-train/src/renderer/tui/progress.rs index 2f8d2a5e1..c4028f319 100644 --- a/crates/burn-train/src/renderer/tui/progress.rs +++ b/crates/burn-train/src/renderer/tui/progress.rs @@ -86,7 +86,7 @@ impl ProgressBarView { let iteration = Gauge::default() .gauge_style(Style::default().fg(Color::Yellow)) - .ratio(self.progress); + .ratio(self.progress.min(1.0)); let eta = Paragraph::new(Line::from(vec![ Span::from(" ("), Span::from(self.eta).italic(),