Update experimental wrappers (#176)

2025-08-22 15:11:51 +00:00 · 2022-12-05 19:14:56 +00:00
parent 1a381bcd0d
commit 848b7097bf
39 changed files with 1140 additions and 806 deletions
--- a/tests/vector/test_vector_env.py
+++ b/tests/vector/test_vector_env.py
@@ -82,8 +82,8 @@ def test_final_obs_info(vectoriser):
        return GenericTestEnv(
            action_space=Discrete(4),
            observation_space=Discrete(4),
-            reset_fn=reset_fn,
-            step_fn=lambda self, action: (
+            reset_func=reset_fn,
+            step_func=lambda self, action: (
                action if action < 3 else 0,
                0,
                action >= 3,