Resolve comments

apache · Aug 15, 2024 · 937c1b3 · 937c1b3
1 parent fd8ed0b
commit 937c1b3
Showing 1 changed file with 7 additions and 2 deletions.
diff --git a/gluten-core/src/main/scala/org/apache/gluten/expression/ExpressionMappings.scala b/gluten-core/src/main/scala/org/apache/gluten/expression/ExpressionMappings.scala
@@ -18,8 +18,9 @@ package org.apache.gluten.expression
 
 import org.apache.gluten.GlutenConfig
 import org.apache.gluten.backendsapi.BackendsApiManager
+import org.apache.gluten.exception.GlutenException
 import org.apache.gluten.expression.ExpressionNames._
-import org.apache.gluten.extension.{DefaultExpressionExtensionTransformer, ExpressionExtensionTrait}
+import org.apache.gluten.extension.ExpressionExtensionTrait
 import org.apache.gluten.sql.shims.SparkShimLoader
 
 import org.apache.spark.sql.catalyst.expressions._
@@ -357,7 +358,11 @@ object ExpressionMappings {
   private var expressionExtensionTransformer: Option[ExpressionExtensionTrait] = None
 
   def getExpressionExtensionTransformer: ExpressionExtensionTrait = {
-    expressionExtensionTransformer.getOrElse(new DefaultExpressionExtensionTransformer)
+    if (expressionExtensionTransformer.isEmpty) {
+      throw new GlutenException(
+        "The expressionExtensionTransformer is not set properly when ini driver or executor")
+    }
+    expressionExtensionTransformer.get
   }
 
   def setExpressionExtensionTransformer(value: ExpressionExtensionTrait): Unit = {