Spaces:
Sleeping
Sleeping
File size: 2,297 Bytes
be5548b |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 |
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,
agent_start_pos=(1,1),
agent_start_dir=0,
):
self.agent_start_pos = agent_start_pos
self.agent_start_dir = agent_start_dir
super().__init__(
grid_size=size,
max_steps=4*size*size,
# Set this to True for maximum speed
see_through_walls=True
)
def _gen_grid(self, width, height):
# Create an empty grid
self.grid = Grid(width, height)
# Generate the surrounding walls
self.grid.wall_rect(0, 0, width, height)
# Place a goal square in the bottom-right corner
self.put_obj(Goal(), width - 2, height - 2)
# Place the agent
if self.agent_start_pos is not None:
self.agent_pos = self.agent_start_pos
self.agent_dir = self.agent_start_dir
else:
self.place_agent()
self.mission = "get to the green goal square"
class EmptyEnv5x5(EmptyEnv):
def __init__(self, **kwargs):
super().__init__(size=5, **kwargs)
class EmptyRandomEnv5x5(EmptyEnv):
def __init__(self):
super().__init__(size=5, agent_start_pos=None)
class EmptyEnv6x6(EmptyEnv):
def __init__(self, **kwargs):
super().__init__(size=6, **kwargs)
class EmptyRandomEnv6x6(EmptyEnv):
def __init__(self):
super().__init__(size=6, agent_start_pos=None)
class EmptyEnv16x16(EmptyEnv):
def __init__(self, **kwargs):
super().__init__(size=16, **kwargs)
register(
id='MiniGrid-Empty-5x5-v0',
entry_point='gym_minigrid.envs:EmptyEnv5x5'
)
register(
id='MiniGrid-Empty-Random-5x5-v0',
entry_point='gym_minigrid.envs:EmptyRandomEnv5x5'
)
register(
id='MiniGrid-Empty-6x6-v0',
entry_point='gym_minigrid.envs:EmptyEnv6x6'
)
register(
id='MiniGrid-Empty-Random-6x6-v0',
entry_point='gym_minigrid.envs:EmptyRandomEnv6x6'
)
register(
id='MiniGrid-Empty-8x8-v0',
entry_point='gym_minigrid.envs:EmptyEnv'
)
register(
id='MiniGrid-Empty-16x16-v0',
entry_point='gym_minigrid.envs:EmptyEnv16x16'
)
|