fix: Fix the UnionExec match branches in CometExecRule (apache#68)

comphead · Feb 21, 2024 · 97ed2b8 · 97ed2b8
1 parent 424733e
commit 97ed2b8
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala
@@ -327,6 +327,8 @@ class CometSparkSessionExtensions
             case Some(nativeOp) =>
               val cometOp = CometUnionExec(u, u.children)
               CometSinkPlaceHolder(nativeOp, u, cometOp)
+            case None =>
+              u
           }
 
         // Native shuffle for Comet operators