From 86ad872cba2f071a52d2eb178e32f4b3c8d10c14 Mon Sep 17 00:00:00 2001 From: Charlie Crane Date: Thu, 12 Dec 2024 11:32:59 +0000 Subject: [PATCH] #2869 - Committing minor changes to base AbstractAgent class before changing branches --- src/primaite/game/agent/interface.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) 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."""