from irlc.ex07.bandits import StationaryBandit, eval_and_plot
from irlc.ex07.simple_agents import BasicAgent
env = StationaryBandit(k=10)                                                                # 10-armed testbed
agents = [BasicAgent(env, epsilon=.1), BasicAgent(env, epsilon=.01)]                        # Create two agents
k = eval_and_plot(env, agents, num_episodes=10, steps=100, max_episodes=10, use_cache=True)    # Fight!