12345678910111213141516171819202122232425262728293031323334 |
- from gym_minigrid.minigrid import *
- from gym_minigrid.register import register
- class EmptyEnv(MiniGridEnv):
- """
- Empty grid environment, no obstacles, sparse reward
- """
- def __init__(self, size=8):
- super().__init__(gridSize=size, maxSteps=3 * size)
- class EmptyEnv6x6(EmptyEnv):
- def __init__(self):
- super().__init__(size=6)
- class EmptyEnv16x16(EmptyEnv):
- def __init__(self):
- super().__init__(size=16)
- register(
- id='MiniGrid-Empty-6x6-v0',
- entry_point='gym_minigrid.envs:EmptyEnv6x6'
- )
- register(
- id='MiniGrid-Empty-8x8-v0',
- entry_point='gym_minigrid.envs:EmptyEnv'
- )
- register(
- id='MiniGrid-Empty-16x16-v0',
- entry_point='gym_minigrid.envs:EmptyEnv16x16'
- )
|