add batch partitioner