Update

[ghstack-poisoned]
pytorch · Jan 30, 2025 · e53150e · e53150e
2 parents 5da2cc2 + c50e2ba
commit e53150e
Showing 1 changed file with 3 additions and 2 deletions.
diff --git 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")