import gym
class myWrapper(gym.Wrapper):
def __init__(self, env):
gym.Wrapper.__init__(self, env)
def reset(self, addition=None, **kwargs):
obs_n = self.env.reset(**kwargs)
return obs_n
def step(self, actions):
obs_n, reward_n, done_n, info_n = self.env.step(actions)
return obs_n, reward_n, done_n, info_n
gym wrapper
最新推荐文章于 2023-05-28 17:04:35 发布