diff --git a/test/test_env.py b/test/test_env.py index 3234915bd34..508d77d7f2a 100644 --- 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)]