Fix: Add MujocoEnvObsRmsPersistence only if obs_norm is enabled

2023-10-24 13:52:30 +02:00 · 2023-10-24 13:52:30 +02:00 · dd4a0eb430
commit dd4a0eb430
parent 58466ebf5d
1 changed files with 2 additions and 1 deletions
--- a/examples/mujoco/mujoco_env.py
+++ b/examples/mujoco/mujoco_env.py
@ -83,5 +83,6 @@ class MujocoEnvFactory(EnvFactory):
            obs_norm=self.obs_norm,
        )
        envs = ContinuousEnvironments(env=env, train_envs=train_envs, test_envs=test_envs)
+        if self.obs_norm:
            envs.set_persistence(MujocoEnvObsRmsPersistence())
        return envs