drop GroupTuples in window exprs

pola-rs · Nov 1, 2021 · 286ccf0 · 286ccf0
1 parent 8a74bb2
commit 286ccf0
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/polars/polars-lazy/src/physical_plan/expressions/window.rs b/polars/polars-lazy/src/physical_plan/expressions/window.rs
@@ -75,6 +75,8 @@ impl PhysicalExpr for WindowExpr {
                 Ok(DataFrame::new_no_checks(cols))
             }
         }?;
+        // drop the group tuples before we do the left join to reduce allocated memory.
+        drop(gb);
 
         // 3. get the join tuples and use them to take the new Series
         let out_column = out.select_at_idx(out.width() - 1).unwrap();