From 825c9ff150acd3165d1a64c933fb396466dc1bb5 Mon Sep 17 00:00:00 2001 From: Lennart Luttkus Date: Mon, 23 Sep 2024 08:38:42 +0200 Subject: [PATCH] Update sb3_test.py to include additional information in the observation variable during reset --- tests/sb3_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/sb3_test.py b/tests/sb3_test.py index 52c8fa7..054c529 100644 --- a/tests/sb3_test.py +++ b/tests/sb3_test.py @@ -23,7 +23,7 @@ def test_can_load_model_snapshot(): inf_env = RobotEnv() model2 = PPO.load(MODEL_PATH, env=inf_env) - obs = inf_env.reset() + obs, info = inf_env.reset() action, _ = model2.predict(obs, deterministic=True) assert action.shape == inf_env.action_space.shape