fix partition requirement

apache · Jun 28, 2021 · 93d2a91 · 93d2a91
1 parent 2735027
commit 93d2a91
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/datafusion/src/physical_plan/windows.rs b/datafusion/src/physical_plan/windows.rs
@@ -414,7 +414,15 @@ impl ExecutionPlan for WindowAggExec {
     }
 
     fn required_child_distribution(&self) -> Distribution {
-        Distribution::UnspecifiedDistribution
+        if self
+            .window_expr()
+            .iter()
+            .all(|expr| expr.partition_by().is_empty())
+        {
+            Distribution::SinglePartition
+        } else {
+            Distribution::UnspecifiedDistribution
+        }
     }
 
     fn with_new_children(