From 5810253844e55da5ba088295d32915fa7a8a111a Mon Sep 17 00:00:00 2001 From: vmoens Date: Tue, 6 Feb 2024 14:50:25 +0000 Subject: [PATCH] tests --- test/test_env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) 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)]