Skip to content

Factory

mighty.mighty_agents.factory #

Factory for creating agents based on config.

get_agent_class #

get_agent_class(agent_type: str) -> MightyAgent

Transforms config keyword for agents to class.

Source code in mighty/mighty_agents/factory.py 
13
14
15
16
17
18
19
20
21
def get_agent_class(agent_type: str) -> MightyAgent:
    """Transforms config keyword for agents to class."""
    agent_class = None
    if agent_type in VALID_AGENT_TYPES:
        agent_class = AGENT_CLASSES[agent_type]
    else:
        raise ValueError(f"Unknown agent_type {agent_type}.")

    return agent_class  # type: ignore