fixes based on PR suggestions
This commit is contained in:
@@ -36,7 +36,7 @@ class AgentLog:
|
|||||||
super().__init__()
|
super().__init__()
|
||||||
self.agent_name = agent_name if agent_name else "unnamed_agent"
|
self.agent_name = agent_name if agent_name else "unnamed_agent"
|
||||||
self.current_timestep: int = 0
|
self.current_timestep: int = 0
|
||||||
self.current_episode: int = 0
|
self.current_episode: int = 1
|
||||||
self.setup_logger()
|
self.setup_logger()
|
||||||
|
|
||||||
@property
|
@property
|
||||||
|
|||||||
@@ -155,7 +155,7 @@ class AbstractAgent(BaseModel, ABC):
|
|||||||
|
|
||||||
@classmethod
|
@classmethod
|
||||||
def from_config(cls, config: Dict) -> AbstractAgent:
|
def from_config(cls, config: Dict) -> AbstractAgent:
|
||||||
"""Grab the relevatn agent class and construct an instance from a config dict."""
|
"""Grab the relevant agent class and construct an instance from a config dict."""
|
||||||
agent_type = config["type"]
|
agent_type = config["type"]
|
||||||
agent_class = cls._registry[agent_type]
|
agent_class = cls._registry[agent_type]
|
||||||
return agent_class(config=config)
|
return agent_class(config=config)
|
||||||
|
|||||||
Reference in New Issue
Block a user