dnrops
/
burn
mirror of https://github.com/tracel-ai/burn.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

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`
This commit is contained in:
Paul Wagener 2024-09-01 22:33:25 +02:00 committed by GitHub
parent 44030ead17
commit 23622d765d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
crates/burn-train/src/renderer/tui/progress.rs
View File

@ -86,7 +86,7 @@ impl ProgressBarView {
let iteration = Gauge::default() let iteration = Gauge::default()
.gauge_style(Style::default().fg(Color::Yellow)) .gauge_style(Style::default().fg(Color::Yellow))
.ratio(self.progress); .ratio(self.progress.min(1.0));
let eta = Paragraph::new(Line::from(vec![ let eta = Paragraph::new(Line::from(vec![
Span::from(" ("), Span::from(" ("),
Span::from(self.eta).italic(), Span::from(self.eta).italic(),