Unwrap env to access viewer

This commit is contained in:
Tom Brown
2017-02-01 15:54:41 -08:00
parent d337f4e571
commit 2d41d698c2

View File

@@ -34,8 +34,8 @@ def key_release(key, mod):
human_agent_action = 0
env.render()
env.viewer.window.on_key_press = key_press
env.viewer.window.on_key_release = key_release
env.unwrapped.viewer.window.on_key_press = key_press
env.unwrapped.viewer.window.on_key_release = key_release
def rollout(env):
global human_agent_action, human_wants_restart, human_sets_pause