diff --git a/src/primaite/game/agent/interface.py b/src/primaite/game/agent/interface.py index 1b9dbcd6..56404e13 100644 --- a/src/primaite/game/agent/interface.py +++ b/src/primaite/game/agent/interface.py @@ -96,8 +96,8 @@ class AbstractAgent(BaseModel, ABC, identifier="Abstract_Agent"): """Base class for scripted and RL agents.""" _registry: ClassVar[Dict[str, Type[AbstractAgent]]] = {} + config: "AbstractAgent.ConfigSchema" - agent_name = "Abstract_Agent" class ConfigSchema(BaseModel): """ @@ -130,6 +130,11 @@ class AbstractAgent(BaseModel, ABC, identifier="Abstract_Agent"): raise ValueError(f"Cannot create a new agent under reserved name {identifier}") cls._registry[identifier] = cls + @property + def logger(self) -> AgentLog: + """Return the AgentLog.""" + return self.config.logger + @property def observation_manager(self) -> ObservationManager: """Returns the agents observation manager."""