def step(self, action): custom_done = False obs, _reward, _done, info = self.env.step(action) self.frame_stack.append(obs[::2, ::2, :])
def step(self, action):
custom_done = False
obs, _reward, _done, info = self.env.step(action)
self.frame_stack.append(obs[::2, ::2, :])