Update (base update)

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