flax

fb3107fd - Use short-circuiting access to debug_key_reuse

Use short-circuiting access to debug_key_reuse Followup to https://github.com/google/flax/pull/3778; because only one or the other attribute exists, we need short-circuiting logic that getattr doesn't have. PiperOrigin-RevId: 618060456