from irlc import plotenv
from irlc.pacman.pacman_environment import PacmanEnvironment, very_small_haunted_maze
env = PacmanEnvironment(layout_str=very_small_haunted_maze, render_mode='human')
s0, _ = env.reset() # Works just like any other environment.
s0 = s0.f("East") # Pacman has now moved east
env.game.state = s0.f("West") # Small hack for making the visualization.
plotenv(env)
env.close()