From d21543eb4fcced5bcf717dca62155d53a4c2bc87 Mon Sep 17 00:00:00 2001
From: Simon Alibert <alibert.sim@gmail.com>
Date: Tue, 9 Apr 2024 10:41:20 +0200
Subject: [PATCH] Add env.close()

---
 tests/test_envs.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/test_envs.py b/tests/test_envs.py
index 72bc93c4..c49461a0 100644
--- a/tests/test_envs.py
+++ b/tests/test_envs.py
@@ -34,7 +34,7 @@ def test_env(env_name, task, obs_type):
     importlib.import_module(package_name)
     env = gym.make(f"{package_name}/{task}", obs_type=obs_type)
     check_env(env.unwrapped)
-
+    env.close()
 
 @pytest.mark.parametrize(
     "env_name",
@@ -61,3 +61,5 @@ def test_factory(env_name):
         # TODO(rcadene): we assume for now that image normalization takes place in the model
         assert img.max() <= 1.0
         assert img.min() >= 0.0
+
+    env.close()