#2829: Update and add nmne tests

src/primaite/game/agent/observations/nic_observations.py

@@ -14,8 +14,9 @@ from primaite.simulator.network.transmission.transport_layer import Port
 
 class NICObservation(AbstractObservation, identifier="NETWORK_INTERFACE"):
     """Status information about a network interface within the simulation environment."""
+
     capture_nmne: ClassVar[bool] = NMNEConfig().capture_nmne
-    "A dataclass defining malicious network events to be captured."
+    "A Boolean specifying whether malicious network events should be captured."
 
     class ConfigSchema(AbstractObservation.ConfigSchema):
         """Configuration schema for NICObservation."""

src/primaite/game/game.py

@@ -9,8 +9,8 @@ from pydantic import BaseModel, ConfigDict
 from primaite import DEFAULT_BANDWIDTH, getLogger
 from primaite.game.agent.actions import ActionManager
 from primaite.game.agent.interface import AbstractAgent, AgentSettings, ProxyAgent
-from primaite.game.agent.observations.observation_manager import ObservationManager
 from primaite.game.agent.observations import NICObservation
+from primaite.game.agent.observations.observation_manager import ObservationManager
 from primaite.game.agent.rewards import RewardFunction, SharedReward
 from primaite.game.agent.scripted_agents.data_manipulation_bot import DataManipulationAgent
 from primaite.game.agent.scripted_agents.probabilistic_agent import ProbabilisticAgent