empty.py 730 B

12345678910111213141516171819202122232425262728293031323334
  1. from gym_minigrid.minigrid import *
  2. from gym_minigrid.register import register
  3. class EmptyEnv(MiniGridEnv):
  4. """
  5. Empty grid environment, no obstacles, sparse reward
  6. """
  7. def __init__(self, size=8):
  8. super().__init__(gridSize=size, maxSteps=3 * size)
  9. class EmptyEnv6x6(EmptyEnv):
  10. def __init__(self):
  11. super().__init__(size=6)
  12. class EmptyEnv16x16(EmptyEnv):
  13. def __init__(self):
  14. super().__init__(size=16)
  15. register(
  16. id='MiniGrid-Empty-6x6-v0',
  17. entry_point='gym_minigrid.envs:EmptyEnv6x6'
  18. )
  19. register(
  20. id='MiniGrid-Empty-8x8-v0',
  21. entry_point='gym_minigrid.envs:EmptyEnv'
  22. )
  23. register(
  24. id='MiniGrid-Empty-16x16-v0',
  25. entry_point='gym_minigrid.envs:EmptyEnv16x16'
  26. )