Skip to content

Commit a2cbdf0

Browse files
committed
Use imported UnsignedPType in take consumers
Signed-off-by: Daniel King <dan@spiraldb.com>
1 parent 72a0cd8 commit a2cbdf0

3 files changed

Lines changed: 9 additions & 6 deletions

File tree

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,7 @@ use crate::arrays::dict::TakeExecute;
2323
use crate::arrays::piecewise_sequence::execute_unit_multiplier_index_arrays;
2424
use crate::arrays::piecewise_sequence::validate_index_ranges;
2525
use crate::builtins::ArrayBuiltins;
26+
use crate::dtype::UnsignedPType;
2627
use crate::executor::ExecutionCtx;
2728
use crate::match_each_integer_ptype;
2829
use crate::match_each_unsigned_integer_ptype;
@@ -125,8 +126,8 @@ fn take_piecewise_bits<S, L>(
125126
output_len: usize,
126127
) -> VortexResult<BitBuffer>
127128
where
128-
S: crate::dtype::UnsignedPType,
129-
L: crate::dtype::UnsignedPType,
129+
S: UnsignedPType,
130+
L: UnsignedPType,
130131
{
131132
validate_index_ranges(source.len(), starts, lengths, output_len)?;
132133

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -27,6 +27,7 @@ use crate::arrays::varbin::VarBinArrayExt;
2727
use crate::dtype::DType;
2828
use crate::dtype::IntegerPType;
2929
use crate::dtype::PType;
30+
use crate::dtype::UnsignedPType;
3031
use crate::executor::ExecutionCtx;
3132
use crate::match_each_unsigned_integer_ptype;
3233
use crate::validity::Validity;
@@ -285,8 +286,8 @@ fn gather_piecewise_varbin<S, L, Offset, NewOffset>(
285286
out_offset_ptype: PType,
286287
) -> VortexResult<GatheredPiecewiseVarBin>
287288
where
288-
S: crate::dtype::UnsignedPType,
289-
L: crate::dtype::UnsignedPType,
289+
S: UnsignedPType,
290+
L: UnsignedPType,
290291
Offset: IntegerPType,
291292
NewOffset: IntegerPType,
292293
{

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -24,6 +24,7 @@ use crate::arrays::piecewise_sequence::execute_unit_multiplier_index_arrays;
2424
use crate::arrays::piecewise_sequence::validate_index_ranges;
2525
use crate::arrays::varbinview::BinaryView;
2626
use crate::buffer::BufferHandle;
27+
use crate::dtype::UnsignedPType;
2728
use crate::executor::ExecutionCtx;
2829
use crate::match_each_integer_ptype;
2930
use crate::match_each_unsigned_integer_ptype;
@@ -140,8 +141,8 @@ fn gather_piecewise_views<S, L>(
140141
output_len: usize,
141142
) -> VortexResult<Buffer<BinaryView>>
142143
where
143-
S: crate::dtype::UnsignedPType,
144-
L: crate::dtype::UnsignedPType,
144+
S: UnsignedPType,
145+
L: UnsignedPType,
145146
{
146147
validate_index_ranges(source.len(), starts, lengths, output_len)?;
147148

0 commit comments

Comments
 (0)