gaspersic/Gymnasium
1
0
Fork 0
mirror of https://github.com/Farama-Foundation/Gymnasium.git synced 2025-08-18 04:49:12 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix seeding issue with NormalizeObservation wrapper testing (#766)

Browse Source
This commit is contained in:
Mark Towers
2023-11-07 13:56:46 +00:00
committed by GitHub
parent 27f8e85051
commit 40f5c7bd42
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/wrappers/vector/test_normalize_observation.py
View File

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