fix test error

apache · Mar 11, 2024 · 41b6156 · 41b6156
1 parent a079046
commit 41b6156
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/SparkPlanExecApiImpl.scala b/backends-velox/src/main/scala/io/glutenproject/backendsapi/velox/SparkPlanExecApiImpl.scala
@@ -227,7 +227,7 @@ class SparkPlanExecApiImpl extends SparkPlanExecApi {
         val hashExpr = new Murmur3Hash(newChild.output)
         val projectList = Seq(Alias(hashExpr, "hash_partition_key")()) ++ newChild.output
         val projectTransformer = ProjectExecTransformer(projectList, newChild)
-        val sortOrder = SortOrder(projectList.head, Ascending)
+        val sortOrder = SortOrder(projectTransformer.output.head, Ascending)
         val sortByHashCode = SortExecTransformer(Seq(sortOrder), global = false, projectTransformer)
         val projectValidationResult = projectTransformer.doValidate()
         val sortValidationResult = sortByHashCode.doValidate()

diff --git a/backends-velox/src/test/scala/io/glutenproject/execution/TestOperator.scala b/backends-velox/src/test/scala/io/glutenproject/execution/TestOperator.scala
@@ -1195,7 +1195,7 @@ class TestOperator extends VeloxWholeStageTransformerSuite {
     // scalastyle:on
 
     withSQLConf("spark.sql.execution.sortBeforeRepartition" -> "false") {
-      runQueryAndCompare(""""SELECT /*+ REPARTITION(3) */
+      runQueryAndCompare("""SELECT /*+ REPARTITION(3) */
                            | l_orderkey, l_partkey FROM lineitem""".stripMargin) {
         df =>
           {