#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/e2e_integration_tests/test_uc2_data_manipulation_scenario.py
+++ b/tests/e2e_integration_tests/test_uc2_data_manipulation_scenario.py
@@ -49,7 +49,7 @@ def test_application_install_uninstall_on_uc2():
        cfg = yaml.safe_load(f)

    env = PrimaiteGymEnv(env_config=cfg)
-    env.agent.flatten_obs = False
+    env.agent.config.flatten_obs = False
    env.reset()

    _, _, _, _, _ = env.step(0)