[MLGO] Update MLRegAlloc Test
This was broken by 5322fb6268208a8fc031fb13573dac9729d05db6. Update the
test to be a little more resilient to flaky failures and to pass after
those changes. We should probably delete this now that we have MIR2Vec,
but punting that for now.
