From d527ad1a365d3aff64200ceba6d11cf376f3919f Mon Sep 17 00:00:00 2001 From: Liang-Chi Hsieh Date: Fri, 20 Dec 2024 00:00:44 -0800 Subject: [PATCH] Revert "fix" This reverts commit 9bad173c7751f105bf3ded2ebc2fed0737d1b909. --- .../scala/org/apache/comet/CometSparkSessionExtensions.scala | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala index c36f5da5e..91f780686 100644 --- a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala +++ b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala @@ -1084,7 +1084,7 @@ class CometSparkSessionExtensions override def apply(plan: SparkPlan): SparkPlan = { val eliminatedPlan = plan transformUp { case ColumnarToRowExec(sparkToColumnar: CometSparkToColumnarExec) => sparkToColumnar.child - case c @ ColumnarToRowExec(child) if child.exists(_.isInstanceOf[CometPlan]) => + case c @ ColumnarToRowExec(child) => val op = CometColumnarToRowExec(child) if (c.logicalLink.isEmpty) { op.unsetTagValue(SparkPlan.LOGICAL_PLAN_TAG)