#2912 - Updates to remaining action refactoring

src/primaite/game/agent/actions/host_nic.py

@@ -1,10 +1,13 @@
 # © Crown-owned copyright 2024, Defence Science and Technology Laboratory UK
 
 from typing import Dict, Optional
+
 from pydantic import BaseModel, ConfigDict
+
 from primaite.game.agent.actions.manager import AbstractAction
 from primaite.interface.request import RequestFormat
 
+
 class HostNICAbstractAction(AbstractAction):
     """
     Abstract base class for NIC actions.
@@ -15,25 +18,12 @@ class HostNICAbstractAction(AbstractAction):
 
     class ConfigSchema(AbstractAction.ConfigSchema):
         """Base Configuration schema for HostNIC actions."""
+
         num_nodes: str
         max_nics_per_node: str
         node_name: str
         nic_num: str
 
-    def __init__(self, manager: "ActionManager", num_nodes: int, max_nics_per_node: int, **kwargs) -> None:
-        """Init method for HostNICAbstractAction.
-
-        :param manager: Reference to the ActionManager which created this action.
-        :type manager: ActionManager
-        :param num_nodes: Number of nodes in the simulation.
-        :type num_nodes: int
-        :param max_nics_per_node: Maximum number of NICs per node.
-        :type max_nics_per_node: int
-        """
-        super().__init__(manager=manager)
-        self.shape: Dict[str, int] = {"node_id": num_nodes, "nic_id": max_nics_per_node}
-        self.verb: str  # define but don't initialise: defends against children classes not defining this
-
     @classmethod
     def form_request(cls, config: ConfigSchema) -> RequestFormat:
         """Return the action formatted as a request which can be ingested by the PrimAITE simulation."""
@@ -41,11 +31,13 @@ class HostNICAbstractAction(AbstractAction):
             return ["do_nothing"]
         return ["network", "node", config.node_name, "network_interface", config.nic_num, cls.verb]
 
+
 class HostNICEnableAction(HostNICAbstractAction):
     """Action which enables a NIC."""
 
     class ConfigSchema(HostNICAbstractAction.ConfigSchema):
         """Configuration schema for HostNICEnableAction."""
+
         verb: str = "enable"
 
 
@@ -54,4 +46,5 @@ class HostNICDisableAction(HostNICAbstractAction):
 
     class ConfigSchema(HostNICAbstractAction.ConfigSchema):
         """Configuration schema for HostNICDisableAction."""
-        verb: str = "disable"
+
+        verb: str = "disable"