Merge pull request #204 from stratosphereips:fix-detection

Fix-detection
stratosphereips · Apr 11, 2024 · 36a6448 · 36a6448
2 parents 8af56c4 + 339db94
commit 36a6448
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 1 deletion.
diff --git a/NetSecGameAgents b/NetSecGameAgents
diff --git a/coordinator.py b/coordinator.py
@@ -375,6 +375,7 @@ def _process_generic_action(self, agent_addr: tuple, action: Action) -> dict:
         action="store",
         required=False,
         type=str,
+        default="env/netsecenv_conf.yaml",
     )
 
     args = parser.parse_args()

diff --git a/env/network_security_game.py b/env/network_security_game.py
@@ -1087,6 +1087,7 @@ def step(self, action:components.Action, action_type='netsecenv')-> components.O
 
             # 1. Perform the action
             if random.random() <= action.type.default_success_p or action_type == 'realworld':
+                self._actions_played.append(action)
                 next_state = self._execute_action(self._current_state, action, action_type=action_type)
             else:
                 logger.info("\tAction NOT sucessful")
+410 −41		agents/agent_utils.py
+44 −22		agents/interactive_tui/assistant.py
+185 −115		agents/interactive_tui/interactive_tui.py
+8 −1		agents/interactive_tui/layout.tcss
+2 −2		agents/llm_qa/llm_agent_qa.py
+5 −6		agents/llm_utils.py
+19 −17		agents/q_learning/q_agent.py
+7 −0		docs/interactive_tui.md
+11 −0		docs/q-learning.md