Make gym_wrapper subclass py_environment.
Make gym_wrapper subclass py_environment.

PiperOrigin-RevId: 251892227
Change-Id: Ia458b45b6cc338cc950ce8b07df087dceb37a031

tensorflow/agents,tensorflow/agents