#2869 - Minor changes trying to make pytest happy

src/primaite/game/agent/interface.py

@@ -51,9 +51,9 @@ class AbstractAgent(BaseModel):
 
     history: List[AgentHistoryItem] = []
     config: "AbstractAgent.ConfigSchema"
-    action_manager: ActionManager
-    observation_manager: ObservationManager
-    reward_function: RewardFunction
+    action_manager: "ActionManager"
+    observation_manager: "ObservationManager"
+    reward_function: "RewardFunction"
 
     class ConfigSchema(BaseModel):
         """

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

@@ -1,11 +1,6 @@
 # © Crown-owned copyright 2025, Defence Science and Technology Laboratory UK
 
 from primaite.game.agent import interface
-from primaite.game.agent.scripted_agents import (
-    abstract_tap,
-    data_manipulation_bot,
-    probabilistic_agent,
-    random_agent,
-)
+from primaite.game.agent.scripted_agents import abstract_tap, data_manipulation_bot, probabilistic_agent, random_agent
 
 __all__ = ("abstract_tap", "data_manipulation_bot", "interface", "probabilistic_agent", "random_agent")

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

@@ -1,7 +1,6 @@
 # © Crown-owned copyright 2025, Defence Science and Technology Laboratory UK
 from typing import Dict, Optional, Tuple
 
-
 from gymnasium.core import ObsType
 
 from primaite.game.agent.scripted_agents.abstract_tap import AbstractTAPAgent

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

@@ -37,6 +37,7 @@ class PeriodicAgent(AbstractScriptedAgent, identifier="Periodic_Agent"):
 
     class ConfigSchema(AbstractScriptedAgent.ConfigSchema):
         """Configuration Schema for Periodic Agent."""
+
         agent_name: str = "Periodic_Agent"
         """Name of the agent."""
 
@@ -57,13 +58,12 @@ class PeriodicAgent(AbstractScriptedAgent, identifier="Periodic_Agent"):
     def start_variance(self) -> int:
         """Returns the deviation around the start step."""
         return self.config.agent_settings.start_settings.variance
-    
+
     @property
     def frequency(self) -> int:
         """Returns the number of timesteps to wait between performing actions."""
         return self.config.agent_settings.start_settings.frequency
 
-
     def _set_next_execution_timestep(self, timestep: int, variance: int) -> None:
         """Set the next execution timestep with a configured random variance.
 

src/primaite/game/game.py

@@ -525,23 +525,22 @@ class PrimaiteGame:
         agents_cfg = cfg.get("agents", [])
 
         for agent_cfg in agents_cfg:
-            agent_ref = agent_cfg["ref"]  # noqa: F841
+            agent_name = agent_cfg["ref"]  # noqa: F841
             agent_type = agent_cfg["type"]
             action_space_cfg = agent_cfg["action_space"]
             observation_space_cfg = agent_cfg["observation_space"]
             reward_function_cfg = agent_cfg["reward_function"]
             agent_settings = agent_cfg["agent_settings"]
 
-            # CREATE AGENT
             agent_config = {
-                "agent_name": agent_ref,
+                "agent_name": agent_name,
                 "action_manager": action_space_cfg,
                 "observation_manager": observation_space_cfg,
                 "reward_function": reward_function_cfg,
                 "agent_settings": agent_settings,
             }
 
-            # new_agent_cfg.update{}
+            # CREATE AGENT
             if agent_type in AbstractAgent._registry:
                 new_agent = AbstractAgent._registry[agent_cfg["type"]].from_config(config=agent_config)
                 # If blue agent is created, add to game.rl_agents