remove task.partition because split already contains partition id.