diff --git a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/stringFunctions.scala b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/stringFunctions.scala index d5f8d3343d8..827e1d75634 100644 --- a/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/stringFunctions.scala +++ b/sql-plugin/src/main/scala/org/apache/spark/sql/rapids/stringFunctions.scala @@ -1346,7 +1346,7 @@ class GpuStringSplitMeta( case class GpuStringSplit(str: Expression, regex: Expression, limit: Expression) extends GpuTernaryExpression with ImplicitCastInputTypes { - override def dataType: DataType = ArrayType(StringType) + override def dataType: DataType = ArrayType(StringType, containsNull = false) override def inputTypes: Seq[DataType] = Seq(StringType, StringType, IntegerType) override def first: Expression = str override def second: Expression = regex