diff --git a/test/test_env.py b/test/test_env.py index 600495a04c1..4f195add548 100644 --- a/test/test_env.py +++ b/test/test_env.py @@ -3285,8 +3285,9 @@ def test_dynamic_rollout(self): RuntimeError, match="The environment specs are dynamic. Call rollout with return_contiguous=False", ): - rollout = env.rollout(4) - rollout = env.rollout(4, return_contiguous=False) + env.rollout(4, return_contiguous=True) + env.rollout(4) + env.rollout(4, return_contiguous=False) check_env_specs(env, return_contiguous=False) @pytest.mark.skipif(not _has_gym, reason="requires gym to be installed")