Skip to content

Commit

Permalink
[Minor] Fix test_compose_action_spec (#2493)
Browse files Browse the repository at this point in the history
Co-authored-by: Louis Faury <louis.faury@helsing.ai>
  • Loading branch information
louisfaury and Louis Faury authored Oct 14, 2024
1 parent be15cab commit a4f83ec
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion test/test_transforms.py
Original file line number Diff line number Diff line change
Expand Up @@ -8694,7 +8694,7 @@ def test_compose_action_spec(self):

# Final check to ensure clean sampling from the action_spec
action = env.rand_action()
assert "action_2"
assert "action_2" in action

@pytest.mark.parametrize("device", get_default_devices())
def test_finitetensordictcheck(self, device):
Expand Down

0 comments on commit a4f83ec

Please sign in to comment.