Fix

apache · Apr 30, 2024 · 82c63c4 · 82c63c4
1 parent 1181dc4
commit 82c63c4
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/spark/src/main/scala/org/apache/comet/serde/QueryPlanSerde.scala b/spark/src/main/scala/org/apache/comet/serde/QueryPlanSerde.scala
@@ -2017,9 +2017,10 @@ object QueryPlanSerde extends Logging with ShimQueryPlanSerde {
    */
   def supportedSortMergeJoinEqualType(dataType: DataType): Boolean = dataType match {
     case _: ByteType | _: ShortType | _: IntegerType | _: LongType | _: FloatType |
-        _: DoubleType | _: StringType | _: DateType | _: TimestampNTZType | _: DecimalType |
-        _: BooleanType =>
+        _: DoubleType | _: StringType | _: DateType | _: DecimalType | _: BooleanType =>
       true
+    // `TimestampNTZType` is private in Spark 3.2/3.3.
+    case dt if dt.typeName == "timestamp_ntz" => true
     case _ => false
   }