`ppo_trainer.update_optimization_state` no longer takes `policy_and_value_opt_state`
`ppo_trainer.update_optimization_state` no longer takes `policy_and_value_opt_state`

PiperOrigin-RevId: 277792375

google/trax,google/trax