def __init__(self, env):
    super(ImageToPyTorch, self).__init__(env)
    old_shape = self.observation_space.shape
    self.observation_space = gym.spaces.Box(low=0, high=255, shape=(old_shape[(- 1)], old_shape[0], old_shape[1]), dtype=np.uint8)
def __init__(self, env):
    super(ImageToPyTorch, self).__init__(env)
    old_shape = self.observation_space.shape
    self.observation_space = gym.spaces.Box(low=0, high=255, shape=(old_shape[(- 1)], old_shape[0], old_shape[1]), dtype=np.uint8)