standalone.py 618 B

1234567891011121314151617181920212223242526272829303132333435363738
  1. #!/usr/bin/env python3
  2. from __future__ import division, print_function
  3. import numpy
  4. import gym
  5. import gym_minigrid
  6. def main():
  7. env = gym.make('MiniGrid-Multi-Room-N6-v0')
  8. env.reset()
  9. # Create a window to render into
  10. renderer = env.render('human')
  11. while True:
  12. env.render('human')
  13. action = 0
  14. obs, reward, done, info = env.step(action)
  15. print('reward=%s' % reward)
  16. if done:
  17. print('done!')
  18. env.reset()
  19. # If the window was closed
  20. if not renderer.window:
  21. break
  22. if __name__ == "__main__":
  23. main()