Fix logger inititialisation in agents

This commit is contained in:
Nick Todd
2025-01-22 10:49:42 +00:00
parent d2d09abcb1
commit 3f94c40434
3 changed files with 49 additions and 7 deletions

View File

@@ -68,6 +68,10 @@ class AbstractAgent(BaseModel, ABC):
)
reward_function: RewardFunction.ConfigSchema = Field(default_factory=lambda: RewardFunction.ConfigSchema())
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.logger: AgentLog = AgentLog(agent_name=self.config.ref)
config: "AbstractAgent.ConfigSchema" = Field(default_factory=lambda: AbstractAgent.ConfigSchema())
logger: AgentLog = AgentLog(agent_name="Abstract_Agent")
@@ -81,6 +85,7 @@ class AbstractAgent(BaseModel, ABC):
def __init_subclass__(cls, identifier: Optional[str] = None, **kwargs: Any) -> None:
super().__init_subclass__(**kwargs)
print("cls identifier:", identifier)
if identifier is None:
return
if identifier in cls._registry:
@@ -157,6 +162,8 @@ class AbstractAgent(BaseModel, ABC):
def from_config(cls, config: Dict) -> AbstractAgent:
"""Grab the relevant agent class and construct an instance from a config dict."""
agent_type = config["type"]
print("agent_type:", agent_type)
print("cls._registry:", cls._registry)
agent_class = cls._registry[agent_type]
return agent_class(config=config)