[GitHub] [spark] WeichenXu123 commented on pull request #40607: [SPARK-42993][ML][CONNECT] Make PyTorch Distributor compatible with Spark Connect

["WeichenXu123 (via GitHub)" <gi...@apache.org>]

WeichenXu123 commented on PR #40607:
URL: https://github.com/apache/spark/pull/40607#issuecomment-1495655168

   Followup tasks:
   
   * We should replace `mapInPandas` with `mapInArrow` for better performance
   * For each spark task, we should save each partition data into local disk (in arrow format) before starting torch process, and provide utility reading methods that torch program (running as a child process) can invoke it to iterate the partition data inside torch program.


-- 
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-unsubscribe@spark.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org


---------------------------------------------------------------------
To unsubscribe, e-mail: reviews-unsubscribe@spark.apache.org
For additional commands, e-mail: reviews-help@spark.apache.org