oxbrbtx/PrimAITE
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

893 - applied changes raised during PR

Browse Source
This commit is contained in:
SunilSamra
2023-06-06 13:12:28 +01:00
parent 5add9d620c
commit 58a87ee0c8
2 changed files with 11 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/primaite/environment/primaite_env.py
View File

@@ -536,14 +536,14 @@ class Primaite(Env):
_action: The action space from the agent
"""
# Convert discrete value back to multidiscrete
multidiscrete_action = self.action_dict[_action]
readable_action = self.action_dict[_action]
action_decision = multidiscrete_action[0]
action_permission = multidiscrete_action[1]
action_source_ip = multidiscrete_action[2]
action_destination_ip = multidiscrete_action[3]
action_protocol = multidiscrete_action[4]
action_port = multidiscrete_action[5]
action_decision = readable_action[0]
action_permission = readable_action[1]
action_source_ip = readable_action[2]
action_destination_ip = readable_action[3]
action_protocol = readable_action[4]
action_port = readable_action[5]
if action_decision == 0:
# It's decided to do nothing