From c50e2babaaa034b0f21c2ac951590cb326e6caa1 Mon Sep 17 00:00:00 2001 From: Vincent Moens Date: Thu, 30 Jan 2025 09:21:45 -0800 Subject: [PATCH] Update (base update) [ghstack-poisoned] --- test/test_env.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) 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")