def step(self, action):
    (obs, reward, done, info) = self.env.step(action)
    self.was_real_done = done
    lives = self.env.unwrapped.ale.lives()
    if ((lives < self.lives) and (lives > 0)):
        done = True
    self.lives = lives
    return (obs, reward, done, info)
def step(self, action):
    (obs, reward, done, info) = self.env.step(action)
    self.was_real_done = done
    lives = self.env.unwrapped.ale.lives()
    if ((lives < self.lives) and (lives > 0)):
        done = True
    self.lives = lives
    return (obs, reward, done, info)