Fixes indexing of actions and rewards in the Atari colab
Fixes indexing of actions and rewards in the Atari colab

PiperOrigin-RevId: 428800059
Change-Id: I3effe327095dc59d749dbb39a5f5fae34f4c5040

google-research/rlds,google-research/rlds