agents.py

# All supported agents
from MonteCarloAgent import MCAgent
from DPAgent import DPAgent

AGENTS_MAP = {
    "MCAgent": MCAgent,
    "DPAgent": DPAgent
}