Make `tff.learning.state_with_new_model_weights` consist with previous changes to make ModelWeights unordered.

PiperOrigin-RevId: 294256825