saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-09-05 20:30:42 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update env_checker.py to avoid closing after checking

Browse source
This commit is contained in:
Quentin Gallouédec 2023-06-01 13:58:22 +02:00 committed by GitHub
parent d63a90cca7
commit 30b3a368e3
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
stable_baselines3/common/env_checker.py
View file

@ -448,9 +448,13 @@ def check_env(env: gym.Env, warn: bool = True, skip_render_check: bool = True) -
except NotImplementedError:
pass
# finally, check that the allows calling `close()` even if the environment was already closed
env.close()
try:
env.close()
except Exception as e:
warnings.warn(f"We recommend to allow calling `close()` even if the environment was already closed.")
# Finally, check that the allows calling `close()` even if the environment was already closed
if env.spec is not None:
new_env = env.spec.make()
new_env.close()
try:
new_env.close()
except Exception as e:
logger.warn(
f"Calling `env.close()` on the closed environment should be allowed, but it raised an exception: {e}"
)