Update

[ghstack-poisoned]

Author: Vincent Moens

test/test_libs.py

@@ -1517,7 +1517,8 @@ def test_gymnasium_autoreset(self, venv):  # noqa
         torch.testing.assert_close(r0["next", "done"], r1["next", "done"])
 
     @implement_for("gym")
-    def test_resetting_strategies(self):
+    @pytest.mark.parametrize("heterogeneous", [False, True])
+    def test_resetting_strategies(self, heterogeneous):
         return
 
     @implement_for("gymnasium", None, "1.0.0")