add logging for partition assignment