wip

matthewgapp · Sep 21, 2023 · 81c3439 · 81c3439
1 parent 86db8cf
commit 81c3439
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/datafusion/core/src/physical_optimizer/coalesce_batches.rs b/datafusion/core/src/physical_optimizer/coalesce_batches.rs
@@ -126,7 +126,6 @@ impl TreeNode for CoalesceContext {
     }
 }
 
-/// implement Display method for `DistributionContext` struct.
 impl fmt::Display for CoalesceContext {
     fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
         let plan_string = get_plan_string(&self.plan);

diff --git a/datafusion/core/src/physical_optimizer/enforce_distribution.rs b/datafusion/core/src/physical_optimizer/enforce_distribution.rs
@@ -1614,6 +1614,11 @@ impl fmt::Display for DistributionContext {
     fn fmt(&self, f: &mut Formatter<'_>) -> fmt::Result {
         let plan_string = get_plan_string(&self.plan);
         write!(f, "plan: {:?}", plan_string)?;
+        write!(
+            f,
+            "has_recursive_ancestor: {:?}",
+            self.has_recursive_ancestor,
+        )?;
         for (idx, child) in self.distribution_onwards.iter().enumerate() {
             if let Some(child) = child {
                 write!(f, "idx:{:?}, exec_tree:{}", idx, child)?;