Minor improvement (#1463)

pola-rs · Sep 30, 2021 · 8e045d5 · 8e045d5
1 parent 0af94e1
commit 8e045d5
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 8 deletions.
diff --git a/polars/polars-arrow/src/kernels/set.rs b/polars/polars-arrow/src/kernels/set.rs
@@ -27,7 +27,7 @@ where
             if truthy {
                 av.extend_from_slice(&values[lower..upper])
             } else {
-                av.extend((lower..upper).map(|_| value))
+                av.extend_from_trusted_len_iter(std::iter::repeat(value).take(upper - lower))
             }
         });
 
@@ -48,7 +48,7 @@ pub fn set_with_mask<T: NativeType>(
         .into_iter()
         .for_each(|(lower, upper, truthy)| {
             if truthy {
-                buf.extend((lower..upper).map(|_| value))
+                buf.extend_from_trusted_len_iter(std::iter::repeat(value).take(upper - lower))
             } else {
                 buf.extend_from_slice(&values[lower..upper])
             }

diff --git a/polars/polars-core/src/utils.rs b/polars/polars-core/src/utils.rs
@@ -452,11 +452,7 @@ macro_rules! static_zip {
 macro_rules! df {
     ($($col_name:expr => $slice:expr), +) => {
         {
-            let mut columns = vec![];
-            $(
-                columns.push(Series::new($col_name, $slice));
-            )+
-            DataFrame::new(columns)
+            DataFrame::new(vec![$(Series::new($col_name, $slice),)+])
         }
     }
 }

diff --git a/polars/polars-io/Cargo.toml b/polars/polars-io/Cargo.toml
@@ -46,7 +46,7 @@ ahash = "0.7"
 num = "^0.4.0"
 dirs = "3.0"
 simdutf8 = {version="0.1", optional=true}
-flate2 = {version = "1", optional=true, --default-features=false}
+flate2 = {version = "1", optional=true, default-features=false}
 
 [package.metadata.docs.rs]
 all-features = true