fix spark

substrait-io · Aug 21, 2024 · 545e204 · 545e204
1 parent 8e4ef88
commit 545e204
Showing 1 changed file with 13 additions and 1 deletion.
diff --git a/spark/src/main/scala/io/substrait/spark/expression/IgnoreNullableAndParameters.scala b/spark/src/main/scala/io/substrait/spark/expression/IgnoreNullableAndParameters.scala
@@ -55,7 +55,11 @@ class IgnoreNullableAndParameters(val typeToMatch: ParameterizedType)
   override def visit(`type`: Type.IntervalYear): Boolean =
     typeToMatch.isInstanceOf[Type.IntervalYear]
 
-  override def visit(`type`: Type.IntervalDay): Boolean = typeToMatch.isInstanceOf[Type.IntervalDay]
+  override def visit(`type`: Type.IntervalDay): Boolean =
+    typeToMatch.isInstanceOf[Type.IntervalDay] || typeToMatch.isInstanceOf[ParameterizedType.IntervalDay]
+
+  override def visit(`type`: Type.IntervalCompound): Boolean =
+    typeToMatch.isInstanceOf[Type.IntervalCompound] || typeToMatch.isInstanceOf[ParameterizedType.IntervalCompound]
 
   override def visit(`type`: Type.UUID): Boolean = typeToMatch.isInstanceOf[Type.UUID]
 
@@ -103,6 +107,14 @@ class IgnoreNullableAndParameters(val typeToMatch: ParameterizedType)
   override def visit(expr: ParameterizedType.Decimal): Boolean =
     typeToMatch.isInstanceOf[Type.Decimal] || typeToMatch.isInstanceOf[ParameterizedType.Decimal]
 
+  @throws[RuntimeException]
+  override def visit(expr: ParameterizedType.IntervalDay): Boolean =
+    typeToMatch.isInstanceOf[Type.IntervalDay] || typeToMatch.isInstanceOf[ParameterizedType.IntervalDay]
+
+  @throws[RuntimeException]
+  override def visit(expr: ParameterizedType.IntervalCompound): Boolean =
+    typeToMatch.isInstanceOf[Type.IntervalCompound] || typeToMatch.isInstanceOf[ParameterizedType.IntervalCompound]
+
   @throws[RuntimeException]
   override def visit(expr: ParameterizedType.Struct): Boolean =
     typeToMatch.isInstanceOf[Type.Struct] || typeToMatch.isInstanceOf[ParameterizedType.Struct]