From dd4a0eb430a874d137df359fba3660b0e82748cd Mon Sep 17 00:00:00 2001
From: Dominik Jain <d.jain@appliedai.de>
Date: Tue, 24 Oct 2023 13:52:30 +0200
Subject: [PATCH] Fix: Add MujocoEnvObsRmsPersistence only if obs_norm is
 enabled

---
 examples/mujoco/mujoco_env.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/examples/mujoco/mujoco_env.py b/examples/mujoco/mujoco_env.py
index 27d6a39..3a48121 100644
--- 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)
-        envs.set_persistence(MujocoEnvObsRmsPersistence())
+        if self.obs_norm:
+            envs.set_persistence(MujocoEnvObsRmsPersistence())
         return envs