fix(rust): LazyFrame.drop_columns overflow issue when columns.len()>s…

…chema.len() (pola-rs#11716)
cmdlineluser · Oct 14, 2023 · 55c9f8f · 55c9f8f
1 parent 0e0daa9
commit 55c9f8f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/crates/polars-plan/src/logical_plan/builder.rs b/crates/polars-plan/src/logical_plan/builder.rs
@@ -401,7 +401,7 @@ impl LogicalPlanBuilder {
     pub fn drop_columns(self, to_drop: PlHashSet<String>) -> Self {
         let schema = try_delayed!(self.0.schema(), &self.0, into);
 
-        let mut output_schema = Schema::with_capacity(schema.len() - to_drop.len());
+        let mut output_schema = Schema::with_capacity(schema.len().saturating_sub(to_drop.len()));
         let columns = schema
             .iter()
             .filter_map(|(col_name, dtype)| {