gym_wrapper.py 文件源码

def __init__(self, env):
        if isinstance(env, six.string_types):
            env = gym.make(env)
        super(GymWrapper, self).__init__(env)
        if isinstance(env.action_space, spaces.MultiDiscrete):
            raise ValueError("Gym environments with MultiDiscrete spaces aren't supported yet.")
        self.observation_space = _to_rf_space(self.env.observation_space)
        self.action_space = _to_rf_space(self.env.action_space)
        self._obs_to_rf = _make_gym2rf_converter(self.observation_space)
        self._action_to_rf = _make_gym2rf_converter(self.action_space)
        self._action_to_gym = _make_rf2gym_converter(self.action_space)
        seed = reinforceflow.get_random_seed()
        if seed and hasattr(self.env, 'seed'):
            self.env.seed(seed)