Skip to content

Commit

Permalink
Add edge case for vectorized environments
Browse files Browse the repository at this point in the history
  • Loading branch information
Toni-SM committed Aug 3, 2024
1 parent d7f738f commit e5f20de
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions tests/torch/test_wrapper_gymnasium.py
Original file line number Diff line number Diff line change
Expand Up @@ -66,6 +66,7 @@ def test_vectorized_env(capsys: pytest.CaptureFixture):
# check methods
for _ in range(2):
observation, info = env.reset()
observation, info = env.reset() # edge case: vectorized environments are autoreset
assert isinstance(observation, torch.Tensor) and observation.shape == torch.Size([num_envs, 3])
assert isinstance(info, Mapping)
for _ in range(3):
Expand Down

0 comments on commit e5f20de

Please sign in to comment.