def step(self, action):
    (ob, reward, done, info) = self.env.step(action)
    self.frames.append(ob)
    return (self._get_ob(), reward, done, info)
def step(self, action):
    (ob, reward, done, info) = self.env.step(action)
    self.frames.append(ob)
    return (self._get_ob(), reward, done, info)