901 - fixed test_single_action_space.py test

src/primaite/acl/access_control_list.py

@@ -80,11 +80,8 @@ class AccessControlList:
              Indicates block if all conditions are satisfied.
         """
         for rule in self.acl:
-            print("loops through rule", rule, isinstance(rule, ACLRule))
             if isinstance(rule, ACLRule):
-                print("finds rule")
                 if self.check_address_match(rule, _source_ip_address, _dest_ip_address):
-                    print("source and dest ip match")
                     if (rule.get_protocol() == _protocol or rule.get_protocol() == "ANY") and (
                         str(rule.get_port()) == str(_port) or rule.get_port() == "ANY"
                     ):

src/primaite/config/training_config.py

@@ -267,7 +267,6 @@ def load(file_path: Union[str, Path], legacy_file: bool = False) -> TrainingConf
     :raises TypeError: When the TrainingConfig object cannot be created
         using the values from the config file read from ``file_path``.
     """
-    print("FILE PATH", file_path)
     if not isinstance(file_path, Path):
         file_path = Path(file_path)
     if file_path.exists():

src/primaite/environment/primaite_env.py

@@ -444,12 +444,11 @@ class Primaite(Env):
             _action: The action space from the agent
         """
         # At the moment, actions are only affecting nodes
-
         if self.training_config.action_type == ActionType.NODE:
             self.apply_actions_to_nodes(_action)
         elif self.training_config.action_type == ActionType.ACL:
             self.apply_actions_to_acl(_action)
-        elif len(self.action_dict[_action]) == 7:  # ACL actions in multidiscrete form have len 6
+        elif len(self.action_dict[_action]) == 7:  # ACL actions in multidiscrete form have len 7
             self.apply_actions_to_acl(_action)
         elif len(self.action_dict[_action]) == 4:  # Node actions in multdiscrete (array) from have len 4
             self.apply_actions_to_nodes(_action)
@@ -1248,7 +1247,6 @@ class Primaite(Env):
 
         # Combine the Node dict and ACL dict
         combined_action_dict = {**acl_action_dict, **new_node_action_dict}
-        print("combined dict", combined_action_dict.items())
         return combined_action_dict
 
     def _create_random_red_agent(self):