Spaces:
Sleeping
Sleeping
File size: 1,007 Bytes
9aed787 |
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 |
from grip_env.environment import GridWorldEnv
from grip_env.layout import BoardLayout
from grip_env.pieces import PIECE_NAMES, COLOUR_NAMES
if __name__ == '__main__':
board1 = BoardLayout(board_size=18, num_pieces=4, shapes=PIECE_NAMES, colours=COLOUR_NAMES, seed=640)
agent_start_pos, target_pos, info = board1.set_board_layout(
target_shape = 'P',
target_colour = 'red',
level = 'easy')
env = GridWorldEnv(render_mode="human", size=18, grid_info=info, agent_pos=agent_start_pos, target_pos=target_pos)
env.reset()
env.render()
for i in range(1000):
# RIGHT, DOWN, LEFT, UP
env.step(0)
env.render()
env.step(0)
env.render()
env.step(0)
env.render()
env.step(1)
env.render()
env.step(1)
env.render()
env.step(2)
env.render()
env.step(2)
env.render()
env.step(3)
env.render()
env.step(3)
env.render()
|