fix partition requirement

apache · Jun 28, 2021 · 723e7bc · 723e7bc
1 parent fac065c
commit 723e7bc
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
@@ -411,7 +411,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(