polars-df 0.13.0 → 0.14.0

data/ext/polars/src/map/dataframe.rs

@@ -158,10 +158,16 @@ where
 {
     let skip = usize::from(first_value.is_some());
     if init_null_count == df.height() {
-        ChunkedArray::full_null("apply", df.height())
+        ChunkedArray::full_null(PlSmallStr::from_static("map"), df.height())
     } else {
         let iter = apply_iter(df, lambda, init_null_count, skip);
-        iterator_to_primitive(iter, init_null_count, first_value, "apply", df.height())
+        iterator_to_primitive(
+            iter,
+            init_null_count,
+            first_value,
+            PlSmallStr::from_static("map"),
+            df.height(),
+        )
     }
 }
 
@@ -174,10 +180,16 @@ pub fn apply_lambda_with_bool_out_type(
 ) -> ChunkedArray<BooleanType> {
     let skip = usize::from(first_value.is_some());
     if init_null_count == df.height() {
-        ChunkedArray::full_null("apply", df.height())
+        ChunkedArray::full_null(PlSmallStr::from_static("map"), df.height())
     } else {
         let iter = apply_iter(df, lambda, init_null_count, skip);
-        iterator_to_bool(iter, init_null_count, first_value, "apply", df.height())
+        iterator_to_bool(
+            iter,
+            init_null_count,
+            first_value,
+            PlSmallStr::from_static("map"),
+            df.height(),
+        )
     }
 }
 
@@ -190,10 +202,16 @@ pub fn apply_lambda_with_utf8_out_type(
 ) -> StringChunked {
     let skip = usize::from(first_value.is_some());
     if init_null_count == df.height() {
-        ChunkedArray::full_null("apply", df.height())
+        ChunkedArray::full_null(PlSmallStr::from_static("map"), df.height())
     } else {
         let iter = apply_iter::<String>(df, lambda, init_null_count, skip);
-        iterator_to_utf8(iter, init_null_count, first_value, "apply", df.height())
+        iterator_to_utf8(
+            iter,
+            init_null_count,
+            first_value,
+            PlSmallStr::from_static("map"),
+            df.height(),
+        )
     }
 }
 
@@ -207,7 +225,10 @@ pub fn apply_lambda_with_list_out_type(
 ) -> RbResult<ListChunked> {
     let skip = usize::from(first_value.is_some());
     if init_null_count == df.height() {
-        Ok(ChunkedArray::full_null("apply", df.height()))
+        Ok(ChunkedArray::full_null(
+            PlSmallStr::from_static("map"),
+            df.height(),
+        ))
     } else {
         let mut iters = get_iters_skip(df, init_null_count + skip);
         let iter = ((init_null_count + skip)..df.height()).map(|_| {
@@ -229,7 +250,14 @@ pub fn apply_lambda_with_list_out_type(
                 Err(e) => panic!("ruby function failed {}", e),
             }
         });
-        iterator_to_list(dt, iter, init_null_count, first_value, "apply", df.height())
+        iterator_to_list(
+            dt,
+            iter,
+            init_null_count,
+            first_value,
+            PlSmallStr::from_static("map"),
+            df.height(),
+        )
     }
 }
 

data/ext/polars/src/map/mod.rs

@@ -28,7 +28,7 @@ fn iterator_to_struct(
     it: impl Iterator<Item = Option<Value>>,
     init_null_count: usize,
     first_value: AnyValue,
-    name: &str,
+    name: PlSmallStr,
     capacity: usize,
 ) -> RbResult<RbSeries> {
     let (vals, flds) = match &first_value {
@@ -89,7 +89,7 @@ fn iterator_to_struct(
         items
             .par_iter()
             .zip(flds)
-            .map(|(av, fld)| Series::new(fld.name(), av))
+            .map(|(av, fld)| Series::new(fld.name().clone(), av))
             .collect::<Vec<_>>()
     });
 
@@ -103,7 +103,7 @@ fn iterator_to_primitive<T>(
     it: impl Iterator<Item = Option<T::Native>>,
     init_null_count: usize,
     first_value: Option<T::Native>,
-    name: &str,
+    name: PlSmallStr,
     capacity: usize,
 ) -> ChunkedArray<T>
 where
@@ -136,7 +136,7 @@ fn iterator_to_bool(
     it: impl Iterator<Item = Option<bool>>,
     init_null_count: usize,
     first_value: Option<bool>,
-    name: &str,
+    name: PlSmallStr,
     capacity: usize,
 ) -> ChunkedArray<BooleanType> {
     // safety: we know the iterators len
@@ -166,7 +166,7 @@ fn iterator_to_object(
     it: impl Iterator<Item = Option<ObjectValue>>,
     init_null_count: usize,
     first_value: Option<ObjectValue>,
-    name: &str,
+    name: PlSmallStr,
     capacity: usize,
 ) -> ObjectChunked<ObjectValue> {
     // safety: we know the iterators len
@@ -196,7 +196,7 @@ fn iterator_to_utf8(
     it: impl Iterator<Item = Option<String>>,
     init_null_count: usize,
     first_value: Option<&str>,
-    name: &str,
+    name: PlSmallStr,
     capacity: usize,
 ) -> StringChunked {
     let first_value = first_value.map(|v| v.to_string());
@@ -229,7 +229,7 @@ fn iterator_to_list(
     it: impl Iterator<Item = Option<Series>>,
     init_null_count: usize,
     first_value: Option<&Series>,
-    name: &str,
+    name: PlSmallStr,
     capacity: usize,
 ) -> RbResult<ListChunked> {
     let mut builder =
@@ -246,7 +246,7 @@ fn iterator_to_list(
             Some(s) => {
                 if s.len() == 0 && s.dtype() != dt {
                     builder
-                        .append_series(&Series::full_null("", 0, dt))
+                        .append_series(&Series::full_null(PlSmallStr::EMPTY, 0, dt))
                         .unwrap()
                 } else {
                     builder.append_series(&s).map_err(RbPolarsErr::from)?