Don't do partitioned if key is given as series

pola-rs · Jan 10, 2025 · c0dc063 · c0dc063
1 parent 99b0b04
commit c0dc063
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/crates/polars-mem-engine/src/planner/lp.rs b/crates/polars-mem-engine/src/planner/lp.rs
@@ -25,7 +25,11 @@ fn partitionable_gb(
         // complex expressions in the group_by itself are also not partitionable
         // in this case anything more than col("foo")
         for key in keys {
-            if (expr_arena).iter(key.node()).count() > 1 {
+            if (expr_arena).iter(key.node()).count() > 1
+                || has_aexpr(key.node(), expr_arena, |ae| {
+                    matches!(ae, AExpr::Literal(LiteralValue::Series(_)))
+                })
+            {
                 return false;
             }
         }