fix: Repartition joins when target_partitions = 1 when required

MazterQyou · MazterQyou · commit 8e60e17fb587 · 2025-10-29T21:56:07.000+04:00
Signed-off-by: Alex Qyoun-ae &lt;4062971+MazterQyou@users.noreply.github.com&gt;
diff --git a/datafusion/core/src/physical_plan/planner.rs b/datafusion/core/src/physical_plan/planner.rs
@@ -923,7 +923,8 @@ impl DefaultPhysicalPlanner {
                         })
                         .collect::<Result<join_utils::JoinOn>>()?;
 
-                    if session_state.config.target_partitions > 1
+                    if (session_state.config.target_partitions > 1
+                        || physical_right.output_partitioning().partition_count() > 1)
                         && session_state.config.repartition_joins
                     {
                         let (left_expr, right_expr) = join_on

Original file line number	Diff line number	Diff line change
`@@ -923,7 +923,8 @@ impl DefaultPhysicalPlanner {`
`923`	`923`	`})`
`924`	`924`	`.collect::<Result<join_utils::JoinOn>>()?;`
`925`	`925`
`926`		`- if session_state.config.target_partitions > 1`
	`926`	`+ if (session_state.config.target_partitions > 1`
	`927`	`+ \|\| physical_right.output_partitioning().partition_count() > 1)`
`927`	`928`	`&& session_state.config.repartition_joins`
`928`	`929`	`{`
`929`	`930`	`let (left_expr, right_expr) = join_on`