From 07186920091c7feddac49d64e2ba8da5d630348c Mon Sep 17 00:00:00 2001 From: Orson Peters Date: Fri, 18 Oct 2024 00:36:10 +0200 Subject: [PATCH] fix nested list order propagation and test --- crates/polars-row/src/encode.rs | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/crates/polars-row/src/encode.rs b/crates/polars-row/src/encode.rs index 363a43b6a9e8..f7128ade36b4 100644 --- a/crates/polars-row/src/encode.rs +++ b/crates/polars-row/src/encode.rs @@ -230,7 +230,7 @@ unsafe fn encode_array(encoder: &Encoder, field: &EncodingField, out: &mut RowsE match encoder { Encoder::List { .. } => { let iter = encoder.list_iter(); - crate::variable::encode_iter(iter, out, &EncodingField::new_unsorted()) + crate::variable::encode_iter(iter, out, field) }, Encoder::Leaf(array) => { match array.dtype() { @@ -374,7 +374,7 @@ fn allocate_rows_buf( row_size_fixed + crate::variable::encoded_len( opt_val, - &EncodingField::new_unsorted(), + &field, ), ); } @@ -383,7 +383,7 @@ fn allocate_rows_buf( for (opt_val, row_length) in iter.zip(lengths.iter_mut()) { *row_length += crate::variable::encoded_len( opt_val, - &EncodingField::new_unsorted(), + &field, ) } } @@ -637,7 +637,7 @@ mod test { let out = out.into_array(); assert_eq!( out.values().iter().map(|v| *v as usize).sum::(), - 82411 + 42774 ); } }