From 97ed2b8a39a2fa30ba611aa8a64c68beaa995cc1 Mon Sep 17 00:00:00 2001
From: wankun <wankun@apache.org>
Date: Wed, 21 Feb 2024 23:48:34 +0800
Subject: [PATCH] fix: Fix the UnionExec match branches in CometExecRule (#68)

---
 .../scala/org/apache/comet/CometSparkSessionExtensions.scala    | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala b/spark/src/main/scala/org/apache/comet/CometSparkSessionExtensions.scala
index f4f56f04f..489504def 100644
--- 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