Fix fallback for concat_ws

apache · Nov 3, 2023 · 8f31ceb · 8f31ceb
1 parent 454bd16
commit 8f31ceb
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/gluten-core/src/main/scala/io/glutenproject/expression/StringExpressionTransformer.scala b/gluten-core/src/main/scala/io/glutenproject/expression/StringExpressionTransformer.scala
@@ -136,8 +136,9 @@ case class ConcatWsTransformer(
   extends ExpressionTransformer {
 
   override def doTransform(args: java.lang.Object): ExpressionNode = {
-    if (!children.head.isInstanceOf[StringLiteralNode]) {
-      throw new UnsupportedOperationException(s"$original not supported yet.")
+    if (!children.head.isInstanceOf[LiteralTransformer]) {
+      throw new UnsupportedOperationException(
+        s"$original only supports literal separator as its first argument.")
     }
     GenericExpressionTransformer(substraitExprName, children, original)
       .doTransform(args)