gym.py 文件源码

def __init__(self, name,
                       log_dir,
                       obs_f = None,
                       reward_f = None,
                       clamp_actions = False,
                       monitor = False):

        self._env = gym.make(name)
        self.log_dir = log_dir
        if monitor:
            self._env = wrappers.Monitor(self._env, log_dir, force=True)

        self.obs_f = obs_f
        self.reward_f = reward_f
        self.clamp_actions = clamp_actions

        self.monitor = monitor