gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-09-01 10:27:43 +00:00
Code Issues Packages Projects Releases Wiki Activity

Allow environment to create a new monitor after closing

Browse Source
This commit is contained in:
Greg Brockman
2016-05-15 16:43:53 -07:00
parent 1d41486480
commit 064f7377d9
2 changed files with 17 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
gym/monitoring/monitor.py
View File

@@ -191,6 +191,8 @@ class Monitor(object):
logger.error('Could not close renderer for %s: %s', key, e) logger.error('Could not close renderer for %s: %s', key, e)
self.enabled = False self.enabled = False
# Remove the env's pointer to this monitor
del self.env._monitor
# Stop tracking this for autoclose # Stop tracking this for autoclose
del monitors[self.monitor_id] del monitors[self.monitor_id]

15
gym/monitoring/tests/test_monitor.py
View File

@@ -45,3 +45,18 @@ def test_video_callable():
pass pass
else: else:
assert False assert False
def test_env_reuse():
with tempdir() as temp:
env = gym.make('CartPole-v0')
env.monitor.start(temp)
env.monitor.close()
env.monitor.start(temp, force=True)
env.reset()
env.step(env.action_space.sample())
env.step(env.action_space.sample())
env.monitor.close()
results = monitor.load_results(temp)
assert results['episode_lengths'] == [2], 'Results: {}'.format(results)