# This file may not be shared/redistributed without permission. Please read copyright notice in the git repo. If this file contains other copyright notices disregard this text.
from irlc import Agent, train, savepdf
from irlc.pacman.pacman_environment import PacmanEnvironment
from irlc.ex03.dp_forward import dp_forward
from irlc.ex03.search_problem import SearchProblem
from irlc.ex03.search_problem import EnsureTerminalSelfTransitionsWrapper
from irlc.ex03.pacman_search import layout2, layout1

if __name__ == "__main__":
    env = PacmanEnvironment(layout_str=layout1, render_mode='human')
    env.reset()
    savepdf("ex03_layout1", env=env)
    env.close()

    env = PacmanEnvironment(layout_str=layout1, render_mode='human')
    env.reset()
    savepdf("ex03_layout2", env=env)
    env.close()