#2869 - Actioning review comments

src/primaite/game/agent/agent_log.py

@@ -21,9 +21,10 @@ class _NotJSONFilter(logging.Filter):
 
 class AgentLog:
     """
-    A Agent Log class is a simple logger dedicated to managing and writing logging updates and information for an agent.
+    An Agent Log class is a simple logger dedicated to managing and writing updates and information for an agent.
 
-    Each log message is written to a file located at: <simulation output directory>/agent_name/agent_name.log
+    Each log message is written to a file located at:
+    <simulation output directory>/agent_name/agent_name.log
     """
 
     def __init__(self, agent_name: Optional[str]):

src/primaite/game/agent/interface.py

@@ -115,6 +115,8 @@ class AbstractAgent(BaseModel):
     @classmethod
     def from_config(cls, config: Dict) -> "AbstractAgent":
         """Creates an agent component from a configuration dictionary."""
+        if config["type"] not in cls._registry:
+            return ValueError(f"Invalid Agent Type: {config['type']}")
         obj = cls(
             config=cls.ConfigSchema(**config["agent_settings"]),
             action_manager=ActionManager.from_config(config["game"], config["action_manager"]),

src/primaite/game/agent/scripted_agents/probabilistic_agent.py

@@ -21,7 +21,7 @@ class ProbabilisticAgent(AbstractScriptedAgent, identifier="ProbabilisticAgent")
     class ConfigSchema(AbstractScriptedAgent.ConfigSchema):
         """Configuration schema for Probabilistic Agent."""
 
-        agent_name: str = "ProbabilisticAgent"
+        type: str = "ProbabilisticAgent"
 
         action_probabilities: Dict[int, float] = None
         """Probability to perform each action in the action map. The sum of probabilities should sum to 1."""