amend

pytorch · Feb 15, 2024 · d84f13e · d84f13e
1 parent 11910e8
commit d84f13e
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/test/test_env.py b/test/test_env.py
@@ -2551,8 +2551,11 @@ def make_env(seed, device=device):
     p_env = ParallelEnv(
         2, [functools.partial(make_env, seed=0), functools.partial(make_env, seed=seed)]
     )
-    r_parallel = p_env.rollout(10, policy)
-    assert not r_parallel.exclude("action").requires_grad
+    try:
+        r_parallel = p_env.rollout(10, policy)
+        assert not r_parallel.exclude("action").requires_grad
+    finally:
+        p_env.close()
 
 
 if __name__ == "__main__":