Update

[ghstack-poisoned]
pytorch · Jan 30, 2025 · 260ac98 · 260ac98
2 parents e29c62c + 423cab5
commit 260ac98
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/test/test_env.py b/test/test_env.py
@@ -3433,7 +3433,9 @@ def test_dynamic_rollout(self):
             RuntimeError,
             match="The environment specs are dynamic. Call rollout with return_contiguous=False",
         ):
-            rollout = env.rollout(4, return_contiguous=True)
+            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")