review comments
fix partitioner for loca mode