def reset(self, **kwargs):
if self.was_real_done:<EOL><INDENT>obs = self.env.reset(**kwargs)<EOL><DEDENT>else:<EOL><INDENT>obs, _, _, _ = self.env.step(<NUM_LIT:0>)<EOL><DEDENT>self.lives = self.env.unwrapped.ale.lives()<EOL>return obs<EOL>
Reset only when lives are exhausted.
        This way all states are still reachable even though lives are episodic,
        and the learner need not know about any of this behind-the-scenes.