Skip to content

Commit

Permalink
Update sb3_test.py to include additional information in the observati…
Browse files Browse the repository at this point in the history
…on variable during reset
  • Loading branch information
ll7 committed Sep 23, 2024
1 parent defd7a5 commit 825c9ff
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/sb3_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down

0 comments on commit 825c9ff

Please sign in to comment.