Revert "Refactor ModelGroup#assign_models to be more performant"

This reverts commit 29156f04a3fdb4822c5f018deed3e6d0d8260ad7.

While this passed tests, we discovered a potential UR bug that kept this
from working correctly. Reverting pending bugfix.