Update tree_node.rs

peter-toth · Feb 29, 2024 · cb7f965 · cb7f965
1 parent 3841067
commit cb7f965
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/datafusion/common/src/tree_node.rs b/datafusion/common/src/tree_node.rs
@@ -644,11 +644,17 @@ impl<T: ConcreteTreeNode> TreeNode for T {
         let (new_self, children) = self.take_children();
         if !children.is_empty() {
             let new_children = children.into_iter().map_until_stop_and_collect(f)?;
-            // Propagate up `t.transformed` and `t.tnr` along with
-            // the node containing transformed children.
-            new_self.with_new_children(new_children.data)
+            if new_children.transformed {
+                // Propagate up `t.transformed` and `t.tnr` along with
+                // the node containing transformed children.
+                new_self.with_new_children(new_children.data)
+            } else {
+                Ok(Transformed::no(
+                    new_self.with_new_children(new_children.data)?.data,
+                ))
+            }
         } else {
-            Ok(Transformed::no(new_self.with_new_children(children)?.data))
+            Ok(Transformed::no(new_self))
         }
     }
 }