Fix default `max_episode_steps` in suite_gym.wrap_env causing errors.
Fix default `max_episode_steps` in suite_gym.wrap_env causing errors.

Fixes #137

PiperOrigin-RevId: 252665992
Change-Id: I758c78cedf03bea18fec901f4ee5d7d2a0b0ed45

tensorflow/agents,tensorflow/agents