fix nested list order propagation and test

pola-rs · Oct 17, 2024 · 0718692 · 0718692
1 parent b05f51c
commit 0718692
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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::<usize>(),
-            82411
+            42774
         );
     }
 }