Remove redundant 'if TYPE_CHECKING' statements

src/primaite/environment/observations.py

@@ -2,6 +2,7 @@
 """Module for handling configurable observation spaces in PrimAITE."""
 import logging
 from abc import ABC, abstractmethod
+from logging import Logger
 from typing import Dict, Final, List, Tuple, TYPE_CHECKING, Union
 
 import numpy as np
@@ -15,12 +16,10 @@ from primaite.nodes.service_node import ServiceNode
 # TYPE_CHECKING is False at runtime and True when typecheckers are performing typechecking
 # Therefore, this avoids circular dependency problem.
 if TYPE_CHECKING:
-    from logging import Logger
-
     from primaite.environment.primaite_env import Primaite
 
 
-_LOGGER: "Logger" = logging.getLogger(__name__)
+_LOGGER: Logger = logging.getLogger(__name__)
 
 
 class AbstractObservationComponent(ABC):

src/primaite/environment/primaite_env.py

@@ -3,9 +3,10 @@
 import copy
 import logging
 import uuid as uuid
+from logging import Logger
 from pathlib import Path
 from random import choice, randint, sample, uniform
-from typing import Any, Dict, Final, List, Tuple, TYPE_CHECKING, Union
+from typing import Any, Dict, Final, List, Tuple, Union
 
 import networkx as nx
 import numpy as np
@@ -49,10 +50,7 @@ from primaite.pol.red_agent_pol import apply_red_agent_iers, apply_red_agent_nod
 from primaite.transactions.transaction import Transaction
 from primaite.utils.session_output_writer import SessionOutputWriter
 
-if TYPE_CHECKING:
-    from logging import Logger
-
-_LOGGER: "Logger" = getLogger(__name__)
+_LOGGER: Logger = getLogger(__name__)
 
 
 class Primaite(Env):

src/primaite/environment/reward.py

@@ -1,5 +1,6 @@
 # Crown Owned Copyright (C) Dstl 2023. DEFCON 703. Shared in confidence.
 """Implements reward function."""
+from logging import Logger
 from typing import Dict, TYPE_CHECKING, Union
 
 from primaite import getLogger
@@ -10,12 +11,10 @@ from primaite.nodes.active_node import ActiveNode
 from primaite.nodes.service_node import ServiceNode
 
 if TYPE_CHECKING:
-    from logging import Logger
-
     from primaite.config.training_config import TrainingConfig
     from primaite.pol.ier import IER
 
-_LOGGER: "Logger" = getLogger(__name__)
+_LOGGER: Logger = getLogger(__name__)
 
 
 def calculate_reward_function(