Use 0 when there are no relearning steps; ensure not NaN

This commit is contained in:
Damien Elmes 2023-10-05 09:55:05 +10:00
parent d7ff99780e
commit 6c1d7a6703

View file

@ -174,13 +174,12 @@ impl Collection {
} }
let mut arr = [0.0; 5]; let mut arr = [0.0; 5];
for (review_kind, group) in group_sec_by_review_kind.iter().enumerate() { for (review_kind, group) in group_sec_by_review_kind.iter().enumerate() {
let average_secs = let average_secs = group.iter().sum::<u32>() as f64 / group.len() as f64 / 1000.0;
if group.is_empty() && review_kind == RevlogReviewKind::Relearning as usize { arr[review_kind] = if average_secs.is_nan() {
30.0 0.0
} else { } else {
group.iter().sum::<u32>() as f64 / group.len() as f64 / 1000.0 average_secs
}; }
arr[review_kind] = average_secs
} }
arr arr
}; };