Fix eleurent#89

StarDustcod3 · Nov 12, 2022 · 9c4b861 · 9c4b861
1 parent 418654e
commit 9c4b861
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/rl_agents/trainer/evaluation.py b/rl_agents/trainer/evaluation.py
@@ -363,7 +363,7 @@ def write_logging(self):
 
     def reset(self, seed=0):
         seed = self.sim_seed + seed if self.sim_seed is not None else None
-        self.observation = self.wrapped_env.reset()
+        self.observation, info = self.wrapped_env.reset()
         self.agent.seed(seed)  # Seed the agent with the main environment seed
         self.agent.reset()