Add RecordVideo wrapper (#2300)

* Add RecordVideo wrapper

* bug fix

* don't change gym's core API

* add test cases

* reformat
This commit is contained in:
Costa Huang
2021-08-18 16:36:40 -04:00
committed by GitHub
parent ee30486bfb
commit 2853ce4797
7 changed files with 140 additions and 7 deletions

View File

@@ -92,13 +92,14 @@ class AsyncVectorEnv(VectorEnv):
self.env_fns = env_fns
self.shared_memory = shared_memory
self.copy = copy
dummy_env = env_fns[0]()
self.metadata = dummy_env.metadata
if (observation_space is None) or (action_space is None):
dummy_env = env_fns[0]()
observation_space = observation_space or dummy_env.observation_space
action_space = action_space or dummy_env.action_space
dummy_env.close()
del dummy_env
dummy_env.close()
del dummy_env
super(AsyncVectorEnv, self).__init__(
num_envs=len(env_fns),
observation_space=observation_space,