#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/integration_tests/game_layer/observations/test_user_observations.py
+++ b/tests/integration_tests/game_layer/observations/test_user_observations.py
@@ -13,7 +13,7 @@ DATA_MANIPULATION_CONFIG = TEST_ASSETS_ROOT / "configs" / "data_manipulation.yam
 def env_with_ssh() -> PrimaiteGymEnv:
    """Build data manipulation environment with SSH port open on router."""
    env = PrimaiteGymEnv(DATA_MANIPULATION_CONFIG)
-    env.agent.flatten_obs = False
+    env.agent.config.agent_settings.flatten_obs = False
    router: Router = env.game.simulation.network.get_node_by_hostname("router_1")
    router.acl.add_rule(ACLAction.PERMIT, src_port=PORT_LOOKUP["SSH"], dst_port=PORT_LOOKUP["SSH"], position=3)
    return env