from irlc.gridworld.gridworld_environments import FrozenLake
from irlc import Agent, interactive
env = FrozenLake(render_mode="human")  # Pass render_mode='human' for visualization.
env, agent = interactive(env, Agent(env))  # For plotting the agents information.
env.reset()     # You always need to call reset
env.plot()      # Plot the environment.
env.close()