Change typehints after mypy analysis

src/primaite/pol/red_agent_pol.py

@@ -4,6 +4,7 @@ from typing import Dict
 
 from networkx import MultiGraph, shortest_path
 
+from primaite import getLogger
 from primaite.acl.access_control_list import AccessControlList
 from primaite.common.custom_typing import NodeUnion
 from primaite.common.enums import HardwareState, NodePOLInitiator, NodePOLType, NodeType, SoftwareState
@@ -13,6 +14,8 @@ from primaite.nodes.node_state_instruction_red import NodeStateInstructionRed
 from primaite.nodes.service_node import ServiceNode
 from primaite.pol.ier import IER
 
+_LOGGER = getLogger(__name__)
+
 _VERBOSE: bool = False
 
 
@@ -270,8 +273,7 @@ def apply_red_agent_node_pol(
                     # Do nothing, service not on this node
                     pass
             else:
-                if _VERBOSE:
-                    print("Node Red Agent PoL not allowed - misconfiguration")
+                _LOGGER.warning("Node Red Agent PoL not allowed - misconfiguration")
 
             # Only apply the PoL if the checks have passed (based on the initiator type)
             if passed_checks:
@@ -292,8 +294,7 @@ def apply_red_agent_node_pol(
                     if isinstance(target_node, ActiveNode) or isinstance(target_node, ServiceNode):
                         target_node.set_file_system_state(state)
             else:
-                if _VERBOSE:
-                    print("Node Red Agent PoL not allowed - did not pass checks")
+                _LOGGER.debug("Node Red Agent PoL not allowed - did not pass checks")
         else:
             # PoL is not valid in this time step
             pass