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

Reply via email to