use blance for partition assignment