Update arrow-select/src/take.rs

Co-authored-by: Raphael Taylor-Davies <[email protected]>
apache · Oct 2, 2023 · 0ef5425 · 0ef5425
1 parent 25229e5
commit 0ef5425
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/arrow-select/src/take.rs b/arrow-select/src/take.rs
@@ -227,8 +227,7 @@ fn take_impl<IndexType: ArrowPrimitiveType>(
             let mut field_type_ids = Vec::with_capacity(fields.len());
             let mut children = Vec::with_capacity(fields.len());
             let values = values.as_any().downcast_ref::<UnionArray>().unwrap();
-            let type_ids = Int8Array::try_new(values.type_ids().clone(), values.nulls().cloned())?;
-            let type_ids = take_impl(&type_ids, indices)?.to_data().buffer(0).into();
+            let type_ids = take_native(values.type_ids(), indices);
             for (type_id, field) in fields.iter() {
                 let values = values.child(type_id);
                 let values = take_impl(values, indices)?;