def __init__(self, env=None):
super(AtariRescale42x42Wrapper, self).__init__(env)
if isinstance(self.observation_space, spaces.Tuple):
self.observation_space = spaces.Tuple([
gym.spaces.Box(0.0, 1.0, [1, 42, 42])
for space in self.env.observation_space.spaces
])
else:
self.observation_space = gym.spaces.Box(0.0, 1.0, [1, 42, 42])
评论列表
文章目录