901 - changed acl current obs from list to numpy.array, changed default ACL list in training_config.py to FALSE, and tried to make test_seeding_and_deterministic_session.py test without fixed reward results
This commit is contained in:
@@ -94,13 +94,13 @@ class TrainingConfig:
|
||||
"Stable Baselines3 learn/eval output verbosity level"
|
||||
|
||||
# Access Control List/Rules
|
||||
apply_implicit_rule: str = True
|
||||
apply_implicit_rule: str = False
|
||||
"User choice to have Implicit ALLOW or DENY."
|
||||
|
||||
implicit_acl_rule: RulePermissionType = RulePermissionType.DENY
|
||||
"ALLOW or DENY implicit firewall rule to go at the end of list of ACL list."
|
||||
|
||||
max_number_acl_rules: int = 0
|
||||
max_number_acl_rules: int = 10
|
||||
"Sets a limit for number of acl rules allowed in the list and environment."
|
||||
|
||||
# Reward values
|
||||
|
||||
@@ -519,30 +519,26 @@ class AccessControlList(AbstractObservationComponent):
|
||||
port_int = self.env.ports_list.index(port) + 2
|
||||
else:
|
||||
_LOGGER.info(f"Port {port} could not be found.")
|
||||
port_int = None
|
||||
|
||||
# Either do the multiply on the obs space
|
||||
# Change the obs to
|
||||
items_to_add = [
|
||||
permission_int,
|
||||
source_ip_int,
|
||||
dest_ip_int,
|
||||
protocol_int,
|
||||
port_int,
|
||||
position,
|
||||
]
|
||||
position = position * 6
|
||||
for item in items_to_add:
|
||||
# print("position", position, "\nitem", int(item))
|
||||
obs.insert(position, int(item))
|
||||
position += 1
|
||||
obs.extend(
|
||||
[
|
||||
permission_int,
|
||||
source_ip_int,
|
||||
dest_ip_int,
|
||||
protocol_int,
|
||||
port_int,
|
||||
position,
|
||||
]
|
||||
)
|
||||
|
||||
else:
|
||||
starting_position = index * 6
|
||||
for placeholder in range(6):
|
||||
obs.insert(starting_position, 0)
|
||||
starting_position += 1
|
||||
obs.extend([0, 0, 0, 0, 0, 0])
|
||||
|
||||
# print("current obs", obs, "\n" ,len(obs))
|
||||
self.current_observation = obs
|
||||
self.current_observation[:] = obs
|
||||
|
||||
def generate_structure(self):
|
||||
"""Return a list of labels for the components of the flattened observation space."""
|
||||
|
||||
Reference in New Issue
Block a user