Fix seeding issue with NormalizeObservation wrapper testing (#766)

This commit is contained in:
Mark Towers
2023-11-07 13:56:46 +00:00
committed by GitHub
parent 27f8e85051
commit 40f5c7bd42

View File

@@ -25,13 +25,15 @@ def test_against_wrapper(
vec_env = SyncVectorEnv([thunk for _ in range(n_envs)])
vec_env = wrappers.vector.NormalizeObservation(vec_env)
vec_env.reset()
vec_env.reset(seed=123)
vec_env.action_space.seed(123)
for _ in range(n_steps):
vec_env.step(vec_env.action_space.sample())
env = wrappers.Autoreset(thunk())
env = wrappers.NormalizeObservation(env)
env.reset()
env.reset(seed=123)
env.action_space.seed(123)
for _ in range(n_envs * n_steps):
env.step(env.action_space.sample())