diff --git a/gluten-substrait/src/main/scala/org/apache/gluten/expression/ExpressionConverter.scala b/gluten-substrait/src/main/scala/org/apache/gluten/expression/ExpressionConverter.scala index cc8d204cd53c..ab2cbbea71d4 100644 --- a/gluten-substrait/src/main/scala/org/apache/gluten/expression/ExpressionConverter.scala +++ b/gluten-substrait/src/main/scala/org/apache/gluten/expression/ExpressionConverter.scala @@ -284,21 +284,21 @@ object ExpressionConverter extends SQLConfHelper with Logging { expr: Expression, attributeSeq: Seq[Attribute], expressionsMap: Map[Class[_], String]): Option[ExpressionTransformer] = { - Option { - expr match { - case pythonUDF: PythonUDF => - replacePythonUDFWithExpressionTransformer(pythonUDF, attributeSeq, expressionsMap) - case scalaUDF: ScalaUDF => - replaceScalaUDFWithExpressionTransformer(scalaUDF, attributeSeq, expressionsMap) - case _ if HiveUDFTransformer.isHiveUDF(expr) => - BackendsApiManager.getSparkPlanExecApiInstance.genHiveUDFTransformer(expr, attributeSeq) - case staticInvoke: StaticInvoke => - replaceStaticInvokeWithExpressionTransformer(staticInvoke, attributeSeq, expressionsMap) - case invoke: Invoke => - replaceInvokeWithExpressionTransformer(invoke, attributeSeq, expressionsMap) - case _ => - null - } + expr match { + case pythonUDF: PythonUDF => + Option(replacePythonUDFWithExpressionTransformer(pythonUDF, attributeSeq, expressionsMap)) + case scalaUDF: ScalaUDF => + Option(replaceScalaUDFWithExpressionTransformer(scalaUDF, attributeSeq, expressionsMap)) + case _ if HiveUDFTransformer.isHiveUDF(expr) => + Option( + BackendsApiManager.getSparkPlanExecApiInstance.genHiveUDFTransformer(expr, attributeSeq)) + case staticInvoke: StaticInvoke => + Option( + replaceStaticInvokeWithExpressionTransformer(staticInvoke, attributeSeq, expressionsMap)) + case invoke: Invoke => + Option(replaceInvokeWithExpressionTransformer(invoke, attributeSeq, expressionsMap)) + case _ => + None } }