From 9bad173c7751f105bf3ded2ebc2fed0737d1b909 Mon Sep 17 00:00:00 2001 From: Liang-Chi Hsieh Date: Wed, 18 Dec 2024 15:25:03 -0800 Subject: [PATCH] fix --- .../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 91f780686..c36f5da5e 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) => + case c @ ColumnarToRowExec(child) if child.exists(_.isInstanceOf[CometPlan]) => val op = CometColumnarToRowExec(child) if (c.logicalLink.isEmpty) { op.unsetTagValue(SparkPlan.LOGICAL_PLAN_TAG)