fix: avoid integer overflow in offsets_to_groups when bigidx is enabl…

…ed (#11901)
pola-rs · Oct 22, 2023 · 7edac49 · 7edac49
1 parent d936f5c
commit 7edac49
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/crates/polars-lazy/src/dsl/list.rs b/crates/polars-lazy/src/dsl/list.rs
@@ -23,11 +23,9 @@ impl IntoListNameSpace for ListNameSpace {
 fn offsets_to_groups(offsets: &[i64]) -> Option<GroupsProxy> {
     let mut start = offsets[0];
     let end = *offsets.last().unwrap();
-    let fits_into_idx = (end - start) <= IdxSize::MAX as i64;
-    if !fits_into_idx {
+    if IdxSize::try_from(end - start).is_err() {
         return None;
     }
-
     let groups = offsets
         .iter()
         .skip(1)