From 8a08719aea6097d8bd1fb46e2fc24bfbb11ad208 Mon Sep 17 00:00:00 2001 From: Angerszhuuuu Date: Fri, 21 Jun 2024 10:08:56 +0800 Subject: [PATCH] Update ArrowEvalPythonUDTFExec.scala --- .../spark/sql/execution/python/ArrowEvalPythonUDTFExec.scala | 3 ++- 1 file changed, 2 insertions(+), 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 index 91f3e05bab9fd..4d10fac2f04a2 100644 --- 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())