fixup

apache · Aug 1, 2024 · b2c8707 · b2c8707
1 parent 811919f
commit b2c8707
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 4 deletions.
diff --git a/backends-velox/src/main/java/org/apache/gluten/utils/VeloxBatchResizerJniWrapper.java b/backends-velox/src/main/java/org/apache/gluten/utils/VeloxBatchResizerJniWrapper.java
@@ -36,5 +36,6 @@ public long handle() {
     return runtime.getHandle();
   }
 
-  public native long create(int minOutputBatchSize, int maxOutputBatchSize, ColumnarBatchInIterator itr);
+  public native long create(
+      int minOutputBatchSize, int maxOutputBatchSize, ColumnarBatchInIterator itr);
 }
diff --git a/backends-velox/src/main/scala/org/apache/gluten/execution/VeloxResizeBatchesExec.scala b/backends-velox/src/main/scala/org/apache/gluten/execution/VeloxResizeBatchesExec.scala
@@ -36,7 +36,10 @@ import scala.collection.JavaConverters._
  * An operator to coalesce input batches by appending the later batches to the one that comes
  * earlier.
  */
-case class VeloxResizeBatchesExec(override val child: SparkPlan, minOutputBatchSize: Int, maxOutputBatchSize: Int)
+case class VeloxResizeBatchesExec(
+    override val child: SparkPlan,
+    minOutputBatchSize: Int,
+    maxOutputBatchSize: Int)
   extends GlutenPlan
   with UnaryExecNode {
 

diff --git a/backends-velox/src/test/scala/org/apache/gluten/execution/TestOperator.scala b/backends-velox/src/test/scala/org/apache/gluten/execution/TestOperator.scala
@@ -837,7 +837,8 @@ class TestOperator extends VeloxWholeStageTransformerSuite with AdaptiveSparkPla
     withSQLConf(
       "spark.gluten.sql.columnar.backend.velox.resizeBatches.shuffleInput" -> "true",
       "spark.gluten.sql.columnar.maxBatchSize" -> "2",
-      "spark.gluten.sql.columnar.backend.velox.resizeBatches.shuffleInput.range" -> s"$minBatchSize~$maxBatchSize"
+      "spark.gluten.sql.columnar.backend.velox.resizeBatches.shuffleInput.range" ->
+        s"$minBatchSize~$maxBatchSize"
     ) {
       val df = runQueryAndCompare(
         "select l_orderkey, sum(l_partkey) as sum from lineitem " +
@@ -861,7 +862,8 @@ class TestOperator extends VeloxWholeStageTransformerSuite with AdaptiveSparkPla
     withSQLConf(
       "spark.gluten.sql.columnar.backend.velox.resizeBatches.shuffleInput" -> "true",
       "spark.gluten.sql.columnar.maxBatchSize" -> "100",
-      "spark.gluten.sql.columnar.backend.velox.resizeBatches.shuffleInput.range" -> s"$minBatchSize~$maxBatchSize"
+      "spark.gluten.sql.columnar.backend.velox.resizeBatches.shuffleInput.range" ->
+        s"$minBatchSize~$maxBatchSize"
     ) {
       val df = runQueryAndCompare(
         "select l_orderkey, sum(l_partkey) as sum from lineitem " +