#2869 - More YAML/test fixes to address failures

2024-12-17 12:47:54 +00:00
parent 3b1b74fb3a
commit 770896200b
7 changed files with 16 additions and 12 deletions
--- a/tests/assets/configs/basic_firewall.yaml
+++ b/tests/assets/configs/basic_firewall.yaml
@@ -60,6 +60,9 @@ agents:
        start_step: 5
        frequency: 4
        variance: 3
+      action_probabilities:
+        0: 0.4
+        1: 0.6

 simulation:
  network:
--- a/tests/assets/configs/dmz_network.yaml
+++ b/tests/assets/configs/dmz_network.yaml
@@ -85,6 +85,9 @@ agents:
        start_step: 5
        frequency: 4
        variance: 3
+      action_probabilities:
+        0: 0.4
+        1: 0.6


 simulation:
--- a/tests/assets/configs/install_and_configure_apps.yaml
+++ b/tests/assets/configs/install_and_configure_apps.yaml
@@ -92,6 +92,9 @@ agents:
    reward_function:
      reward_components:
        - type: DUMMY
+    agent_settings:
+      flatten_obs: True
+      action_masking: False

 simulation:
  network: