Change the MapPartition api func