agents.py

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

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