viirya commented on a change in pull request #33559: URL: https://github.com/apache/spark/pull/33559#discussion_r786378908
########## File path: sql/core/src/main/scala/org/apache/spark/sql/execution/python/ArrowPythonRunner.scala ########## @@ -96,6 +101,8 @@ class ArrowPythonRunner( writer.writeBatch() arrowWriter.reset() } + val deltaData = dataOut.size() - startData + pythonDataSent += deltaData Review comment: If you intend to measure data output size during writing to python worker, should you move the two lines into the loop so it can keep tracking output size after each batch? -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: reviews-unsubscr...@spark.apache.org For additional commands, e-mail: reviews-h...@spark.apache.org