Apply suggestions from PR review.

2024-03-04 09:58:57 +00:00
parent a6031d568d
commit d1480e4477
16 changed files with 25 additions and 19 deletions
--- a/tests/assets/configs/bad_primaite_session.yaml
+++ b/tests/assets/configs/bad_primaite_session.yaml
@@ -21,7 +21,7 @@ game:
 agents:
  - ref: client_2_green_user
    team: GREEN
-    type: probabilistic_agent
+    type: ProbabilisticAgent
    observation_space:
      type: UC2GreenObservation
    action_space:
--- a/tests/assets/configs/basic_firewall.yaml
+++ b/tests/assets/configs/basic_firewall.yaml
@@ -40,7 +40,7 @@ game:
 agents:
  - ref: client_2_green_user
    team: GREEN
-    type: probabilistic_agent
+    type: ProbabilisticAgent
    observation_space:
      type: UC2GreenObservation
    action_space:
--- a/tests/assets/configs/basic_switched_network.yaml
+++ b/tests/assets/configs/basic_switched_network.yaml
@@ -40,7 +40,7 @@ game:
 agents:
  - ref: client_2_green_user
    team: GREEN
-    type: probabilistic_agent
+    type: ProbabilisticAgent
    observation_space:
      type: UC2GreenObservation
    action_space:
--- a/tests/assets/configs/dmz_network.yaml
+++ b/tests/assets/configs/dmz_network.yaml
@@ -65,7 +65,7 @@ game:
 agents:
  - ref: client_1_green_user
    team: GREEN
-    type: probabilistic_agent
+    type: ProbabilisticAgent
    observation_space:
      type: UC2GreenObservation
    action_space:
--- a/tests/assets/configs/eval_only_primaite_session.yaml
+++ b/tests/assets/configs/eval_only_primaite_session.yaml
@@ -25,7 +25,7 @@ game:
 agents:
  - ref: client_2_green_user
    team: GREEN
-    type: probabilistic_agent
+    type: ProbabilisticAgent
    observation_space:
      type: UC2GreenObservation
    action_space:
--- a/tests/assets/configs/multi_agent_session.yaml
+++ b/tests/assets/configs/multi_agent_session.yaml
@@ -31,7 +31,7 @@ game:
 agents:
  - ref: client_2_green_user
    team: GREEN
-    type: probabilistic_agent
+    type: ProbabilisticAgent
    observation_space:
      type: UC2GreenObservation
    action_space:
--- a/tests/assets/configs/test_primaite_session.yaml
+++ b/tests/assets/configs/test_primaite_session.yaml
@@ -29,7 +29,7 @@ game:
 agents:
  - ref: client_2_green_user
    team: GREEN
-    type: probabilistic_agent
+    type: ProbabilisticAgent
    observation_space:
      type: UC2GreenObservation
    action_space:
--- a/tests/assets/configs/train_only_primaite_session.yaml
+++ b/tests/assets/configs/train_only_primaite_session.yaml
@@ -25,7 +25,7 @@ game:
 agents:
  - ref: client_2_green_user
    team: GREEN
-    type: probabilistic_agent
+    type: ProbabilisticAgent
    observation_space:
      type: UC2GreenObservation
    action_space:
--- a/tests/integration_tests/game_layer/test_rewards.py
+++ b/tests/integration_tests/game_layer/test_rewards.py
@@ -41,6 +41,7 @@ def test_WebpageUnavailablePenalty(game_and_agent):


 def test_uc2_rewards(game_and_agent):
+    """Test that the reward component correctly applies a penalty when the selected client cannot reach the database."""
    game, agent = game_and_agent
    agent: ControlledAgent