diff --git a/test/test_env.py b/test/test_env.py index 7e0c4f387f8..d215f858fa6 100644 --- a/test/test_env.py +++ b/test/test_env.py @@ -3290,8 +3290,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")