#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.