fix(rust): Ensure ListPrimitiveBuilder dtype invariant is asserted (#…

…18889)
pola-rs · Sep 24, 2024 · e9d835d · e9d835d
1 parent afff7a4
commit e9d835d
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/crates/polars-core/src/chunked_array/builder/list/primitive.rs b/crates/polars-core/src/chunked_array/builder/list/primitive.rs
@@ -17,11 +17,15 @@ where
         name: PlSmallStr,
         capacity: usize,
         values_capacity: usize,
-        logical_type: DataType,
+        inner_type: DataType,
     ) -> Self {
+        assert!(
+            inner_type.is_numeric() || inner_type.is_temporal(),
+            "inner type must be primitive"
+        );
         let values = MutablePrimitiveArray::<T::Native>::with_capacity(values_capacity);
         let builder = LargePrimitiveBuilder::<T::Native>::new_with_capacity(values, capacity);
-        let field = Field::new(name, DataType::List(Box::new(logical_type)));
+        let field = Field::new(name, DataType::List(Box::new(inner_type)));
 
         Self {
             builder,