Skip to content

Commit dbe40e0

Browse files
committed
[Fix] Add printOutputColumns parameter to generateTreeString methods
see apache/spark#50852
1 parent ad3f405 commit dbe40e0

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

shims/spark41/src/main/scala/org/apache/gluten/execution/GenerateTreeStringShim.scala

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -42,6 +42,7 @@ trait WholeStageTransformerGenerateTreeStringShim extends UnaryExecNode {
4242
addSuffix: Boolean = false,
4343
maxFields: Int,
4444
printNodeId: Boolean,
45+
printOutputColumns: Boolean,
4546
indent: Int = 0): Unit = {
4647
val prefix = if (printNodeId) "^ " else s"^($stageId) "
4748
child.generateTreeString(
@@ -53,7 +54,8 @@ trait WholeStageTransformerGenerateTreeStringShim extends UnaryExecNode {
5354
addSuffix = false,
5455
maxFields,
5556
printNodeId = printNodeId,
56-
indent)
57+
printOutputColumns = printOutputColumns,
58+
indent = indent)
5759

5860
if (verbose && wholeStageTransformerContextDefined) {
5961
append(prefix + "Substrait plan:\n")
@@ -74,6 +76,7 @@ trait InputAdapterGenerateTreeStringShim extends UnaryExecNode {
7476
addSuffix: Boolean = false,
7577
maxFields: Int,
7678
printNodeId: Boolean,
79+
printOutputColumns: Boolean,
7780
indent: Int = 0): Unit = {
7881
child.generateTreeString(
7982
depth,
@@ -84,6 +87,7 @@ trait InputAdapterGenerateTreeStringShim extends UnaryExecNode {
8487
addSuffix = false,
8588
maxFields,
8689
printNodeId,
90+
printOutputColumns,
8791
indent)
8892
}
8993
}

0 commit comments

Comments
 (0)