fix sac obs error (#259)

Denys88 · Oct 17, 2023 · 33ba628 · 33ba628
1 parent fe95913
commit 33ba628
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/rl_games/algos_torch/sac_agent.py b/rl_games/algos_torch/sac_agent.py
@@ -493,8 +493,9 @@ def play_steps(self, random_exploration = False):
 
             if isinstance(next_obs, dict):    
                 next_obs_processed = next_obs['obs']
-
-            self.obs = next_obs.clone()
+                self.obs = next_obs_processed.clone()
+            else:
+                self.obs = next_obs.clone()
 
             rewards = self.rewards_shaper(rewards)