Import UnsignedPType in take helpers

Signed-off-by: Daniel King <dan@spiraldb.com>

Author: danking

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

@@ -23,6 +23,7 @@ use crate::arrays::dict::TakeExecute;
 use crate::arrays::piecewise_sequence::execute_unit_multiplier_index_arrays;
 use crate::arrays::piecewise_sequence::validate_index_ranges;
 use crate::builtins::ArrayBuiltins;
+use crate::dtype::UnsignedPType;
 use crate::executor::ExecutionCtx;
 use crate::match_each_integer_ptype;
 use crate::match_each_unsigned_integer_ptype;
@@ -125,8 +126,8 @@ fn take_piecewise_bits<S, L>(
     output_len: usize,
 ) -> VortexResult<BitBuffer>
 where
-    S: crate::dtype::UnsignedPType,
-    L: crate::dtype::UnsignedPType,
+    S: UnsignedPType,
+    L: UnsignedPType,
 {
     validate_index_ranges(source.len(), starts, lengths, output_len)?;
 

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

@@ -18,6 +18,7 @@ use crate::arrays::piecewise_sequence::execute_unit_multiplier_index_arrays;
 use crate::arrays::piecewise_sequence::validate_index_ranges;
 use crate::dtype::IntegerPType;
 use crate::dtype::NativeDecimalType;
+use crate::dtype::UnsignedPType;
 use crate::executor::ExecutionCtx;
 use crate::match_each_decimal_value_type;
 use crate::match_each_integer_ptype;
@@ -95,8 +96,8 @@ fn take_piecewise_to_buffer<S, L, T>(
     output_len: usize,
 ) -> VortexResult<Buffer<T>>
 where
-    S: crate::dtype::UnsignedPType,
-    L: crate::dtype::UnsignedPType,
+    S: UnsignedPType,
+    L: UnsignedPType,
     T: NativeDecimalType,
 {
     validate_index_ranges(values.len(), starts, lengths, output_len)?;

vortex-array/src/arrays/piecewise_sequence/vtable.rs

@@ -35,6 +35,7 @@ use crate::arrays::primitive::PrimitiveArrayExt;
 use crate::buffer::BufferHandle;
 use crate::dtype::DType;
 use crate::dtype::PType;
+use crate::dtype::UnsignedPType;
 use crate::match_each_unsigned_integer_ptype;
 use crate::scalar::Scalar;
 use crate::serde::ArrayChildren;
@@ -245,9 +246,9 @@ fn scalar_at<S, L, M>(
     index: usize,
 ) -> VortexResult<u64>
 where
-    S: crate::dtype::UnsignedPType,
-    L: crate::dtype::UnsignedPType,
-    M: crate::dtype::UnsignedPType,
+    S: UnsignedPType,
+    L: UnsignedPType,
+    M: UnsignedPType,
 {
     let mut remaining = index;
     for ((&start, &length), &multiplier) in starts

vortex-array/src/arrays/primitive/compute/take/mod.rs

@@ -26,6 +26,7 @@ use crate::arrays::piecewise_sequence::validate_index_ranges;
 use crate::builtins::ArrayBuiltins;
 use crate::dtype::DType;
 use crate::dtype::IntegerPType;
+use crate::dtype::UnsignedPType;
 use crate::executor::ExecutionCtx;
 use crate::match_each_integer_ptype;
 use crate::match_each_native_ptype;
@@ -188,8 +189,8 @@ fn primitive_piecewise_values<T, S, L>(
 ) -> VortexResult<Buffer<T>>
 where
     T: Copy,
-    S: crate::dtype::UnsignedPType,
-    L: crate::dtype::UnsignedPType,
+    S: UnsignedPType,
+    L: UnsignedPType,
 {
     validate_index_ranges(source.len(), starts, lengths, output_len)?;
 

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

@@ -27,6 +27,7 @@ use crate::arrays::varbin::VarBinArrayExt;
 use crate::dtype::DType;
 use crate::dtype::IntegerPType;
 use crate::dtype::PType;
+use crate::dtype::UnsignedPType;
 use crate::executor::ExecutionCtx;
 use crate::match_each_unsigned_integer_ptype;
 use crate::validity::Validity;
@@ -285,8 +286,8 @@ fn gather_piecewise_varbin<S, L, Offset, NewOffset>(
     out_offset_ptype: PType,
 ) -> VortexResult<GatheredPiecewiseVarBin>
 where
-    S: crate::dtype::UnsignedPType,
-    L: crate::dtype::UnsignedPType,
+    S: UnsignedPType,
+    L: UnsignedPType,
     Offset: IntegerPType,
     NewOffset: IntegerPType,
 {

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

@@ -24,6 +24,7 @@ use crate::arrays::piecewise_sequence::execute_unit_multiplier_index_arrays;
 use crate::arrays::piecewise_sequence::validate_index_ranges;
 use crate::arrays::varbinview::BinaryView;
 use crate::buffer::BufferHandle;
+use crate::dtype::UnsignedPType;
 use crate::executor::ExecutionCtx;
 use crate::match_each_integer_ptype;
 use crate::match_each_unsigned_integer_ptype;
@@ -140,8 +141,8 @@ fn gather_piecewise_views<S, L>(
     output_len: usize,
 ) -> VortexResult<Buffer<BinaryView>>
 where
-    S: crate::dtype::UnsignedPType,
-    L: crate::dtype::UnsignedPType,
+    S: UnsignedPType,
+    L: UnsignedPType,
 {
     validate_index_ranges(source.len(), starts, lengths, output_len)?;