Currently, IM is part of the partitioner. We need to put that in IM.
Currently, IM is part of the partitioner. We need to put that in IM.