def __init__(self, name, dump_dir=None, force_dump=False, state_mode='DEFAULT'):
super().__init__()
with get_env_lock():
self._gym = self._make_env(name)
if dump_dir:
io.mkdir(dump_dir)
self._gym = gym.wrappers.Monitor(self._gym, dump_dir, force=force_dump)
assert state_mode in ('DEFAULT', 'RENDER', 'BOTH')
self._state_mode = state_mode
评论列表
文章目录