wip

Signed-off-by: Alexander Droste <[email protected]>

Author: 0ax1

vortex-array/src/arrays/list/compute/take.rs

@@ -43,18 +43,17 @@ impl TakeKernel for ListVTable {
             match_each_integer_ptype!(indices.ptype(), |I| {
                 let offsets_slice = offsets.as_slice::<O>();
                 let indices_slice: &[I] = indices.as_slice::<I>();
-
-                // Calculate total count to determine appropriate accumulation type
-                let total_count = indices_slice
+                let approx_total_count = indices_slice
                     .iter()
                     .map(|idx| {
                         let idx: usize = idx.as_();
-                        let diff: usize = (offsets_slice[idx + 1] - offsets_slice[idx]).as_();
-                        diff
+                        let length: usize = (offsets_slice[idx + 1] - offsets_slice[idx]).as_();
+                        length
                     })
-                    .sum::<usize>();
+                    .max()
+                    .unwrap_or(0);
 
-                match_smallest_offset_type!(total_count, |AccumType| {
+                match_smallest_offset_type!(approx_total_count, |AccumType| {
                     _take::<I, O, AccumType>(
                         array,
                         offsets_slice,