ease branch prediction

pola-rs · Apr 19, 2024 · 01064c3 · 01064c3
1 parent 7b68752
commit 01064c3
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/crates/polars-core/src/series/any_value.rs b/crates/polars-core/src/series/any_value.rs
@@ -494,11 +494,13 @@ fn any_values_to_list(
     #[allow(unused_mut)]
     let mut out: ListChunked = if inner_type == &DataType::Null {
         let mut first_value_dtype = DataType::Null;
+        let mut set_first = false;
         avs.iter()
             .map(|av| match av {
                 AnyValue::List(b) => {
-                    if !b.is_empty() && matches!(first_value_dtype, DataType::Null) {
+                    if !set_first && first_value_dtype != *b.dtype() {
                         first_value_dtype = b.dtype().clone();
+                        set_first = true;
                     } else if b.is_empty() || b.null_count() == b.len() {
                         return b.cast(&first_value_dtype).ok();
                     }