Update ArrowEvalPythonUDTFExec.scala

AngersZhuuuu · Jun 21, 2024 · 8a08719 · 8a08719
1 parent 4fecae0
commit 8a08719
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/sql/core/src/main/scala/org/apache/spark/sql/execution/python/ArrowEvalPythonUDTFExec.scala b/sql/core/src/main/scala/org/apache/spark/sql/execution/python/ArrowEvalPythonUDTFExec.scala
@@ -83,7 +83,8 @@ case class ArrowEvalPythonUDTFExec(
       val actualDataTypes = (0 until flattenedBatch.numCols()).map(
         i => flattenedBatch.column(i).dataType())
       val arrowOutputTypes = outputTypes.map(ArrowUtils.toArrowOutputSchema)
-      assert(arrowOutputTypes == actualDataTypes, "Invalid schema from arrow-enabled Python UDTF: " +
+      assert(arrowOutputTypes == actualDataTypes,
+        "Invalid schema from arrow-enabled Python UDTF: " +
         s"expected ${outputTypes.mkString(", ")}, got ${actualDataTypes.mkString(", ")}")
 
       flattenedBatch.setNumRows(batch.numRows())