File tree Expand file tree Collapse file tree 1 file changed +5
-2
lines changed
arrow/src/compute/kernels Expand file tree Collapse file tree 1 file changed +5
-2
lines changed Original file line number Diff line number Diff line change 1818//! Defines windowing functions, like `shift`ing
1919
2020use crate :: array:: { Array , ArrayRef } ;
21- use crate :: { array:: new_null_array, compute:: concat} ;
2221use crate :: { array:: PrimitiveArray , datatypes:: ArrowPrimitiveType , error:: Result } ;
22+ use crate :: {
23+ array:: { make_array, new_null_array} ,
24+ compute:: concat,
25+ } ;
2326use num:: { abs, clamp} ;
2427
2528/// Shifts array by defined number of items (to left or right)
5962{
6063 let value_len = values. len ( ) as i64 ;
6164 if offset == 0 {
62- Ok ( values. slice ( 0 , values . len ( ) ) )
65+ Ok ( make_array ( values. data_ref ( ) . clone ( ) ) )
6366 } else if offset == i64:: MIN || abs ( offset) >= value_len {
6467 Ok ( new_null_array ( & T :: DATA_TYPE , values. len ( ) ) )
6568 } else {
You can’t perform that action at this time.
0 commit comments