diff --git a/rust-toolchain.toml b/rust-toolchain.toml index 76a06e6b8..f25b5b140 100644 --- a/rust-toolchain.toml +++ b/rust-toolchain.toml @@ -1,2 +1,2 @@ [toolchain] -channel = "1.94.0" +channel = "1.95.0" diff --git a/rust/lancedb/src/dataloader/permutation/shuffle.rs b/rust/lancedb/src/dataloader/permutation/shuffle.rs index 7cd27e342..b26db7bea 100644 --- a/rust/lancedb/src/dataloader/permutation/shuffle.rs +++ b/rust/lancedb/src/dataloader/permutation/shuffle.rs @@ -464,11 +464,9 @@ mod tests { let mut iter = ids.into_iter().map(|o| o.unwrap()); while let Some(first) = iter.next() { let rows_left_in_clump = if first == 4470 { 19 } else { 29 }; - let mut expected_next = first + 1; - for _ in 0..rows_left_in_clump { + for expected_next in (first + 1)..=(first + rows_left_in_clump) { let next = iter.next().unwrap(); assert_eq!(next, expected_next); - expected_next += 1; } } } diff --git a/rust/lancedb/src/table.rs b/rust/lancedb/src/table.rs index 0866b3e55..96f0cffcf 100644 --- a/rust/lancedb/src/table.rs +++ b/rust/lancedb/src/table.rs @@ -3015,11 +3015,12 @@ impl BaseTable for NativeTable { let p99 = *sorted_sizes.get(num_fragments * 99 / 100).unwrap_or(&0); let min = sorted_sizes.first().copied().unwrap_or(0); let max = sorted_sizes.last().copied().unwrap_or(0); - let mean = if num_fragments == 0 { - 0 - } else { - sorted_sizes.iter().copied().sum::() / num_fragments - }; + let mean = sorted_sizes + .iter() + .copied() + .sum::() + .checked_div(num_fragments) + .unwrap_or(0); let frag_stats = FragmentStatistics { num_fragments,