flax

204d975c - Typo in exception messages

Typo in exception messages FUTURE_COPYBARA_INTEGRATE_REVIEW=https://github.com/google/flax/pull/797 from afqueiruga:master ae40131c4f991bb8dc0fa757d52319639389b742 PiperOrigin-RevId: 351376151