def make_env(env_name, monitor_dir): env = HistoryWrapper(HISTORY_STEPS)(gym.make(env_name)) if monitor_dir: env = gym.wrappers.Monitor(env, monitor_dir) return env