Split by partitions instead of collect