Add mapPartitionsWithSplit() to PySpark.