From 82c63c4cf9d11e337b12cde7365e26a2ea5a4a59 Mon Sep 17 00:00:00 2001 From: Liang-Chi Hsieh Date: Tue, 30 Apr 2024 07:05:27 -0700 Subject: [PATCH] Fix --- .../main/scala/org/apache/comet/serde/QueryPlanSerde.scala | 5 +++-- 1 file changed, 3 insertions(+), 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 index 45add0ef5..6ebb7190f 100644 --- 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 }