tests

pytorch · Feb 6, 2024 · 5810253 · 5810253
1 parent e02feef
commit 5810253
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_env.py b/test/test_env.py
@@ -629,7 +629,7 @@ def test_parallel_env_with_policy(
     def test_transform_env_transform_no_device(self, heterogeneous):
         # Tests non-regression on 1865
         def make_env():
-            return TransformedEnv(ContinuousActionVecMockEnv(), StepCounter(max_step=3))
+            return TransformedEnv(ContinuousActionVecMockEnv(), StepCounter(max_steps=3))
 
         if heterogeneous:
             make_envs = [EnvCreator(make_env), EnvCreator(make_env)]