also expand rename in filters (#3008)

pola-rs · Mar 30, 2022 · 5387319 · 5387319
1 parent 82c71db
commit 5387319
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/polars/polars-lazy/src/logical_plan/builder.rs b/polars/polars-lazy/src/logical_plan/builder.rs
@@ -274,7 +274,9 @@ impl LogicalPlanBuilder {
 
     /// Apply a filter
     pub fn filter(self, predicate: Expr) -> Self {
-        let predicate = if has_expr(&predicate, |e| matches!(e, Expr::Wildcard)) {
+        let predicate = if has_expr(&predicate, |e| {
+            matches!(e, Expr::Wildcard | Expr::RenameAlias { .. })
+        }) {
             let rewritten = rewrite_projections(vec![predicate], self.0.schema(), &[]);
             combine_predicates_expr(rewritten.into_iter())
         } else {