Fix Bug in Multiprocess Environment

MushroomRL · Oct 15, 2024 · 550971f · 550971f
1 parent de95b84
commit 550971f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mushroom_rl/core/multiprocess_environment.py b/mushroom_rl/core/multiprocess_environment.py
@@ -107,7 +107,7 @@ def reset_all(self, env_mask, state=None):
             else:
                 episode_infos.append({})
 
-        return self._states, episode_infos
+        return self._states.copy(), episode_infos.copy()
 
     def step_all(self, env_mask, action):
         for i, remote in enumerate(self._remotes):
@@ -129,7 +129,7 @@ def step_all(self, env_mask, action):
             else:
                 step_infos.append({})
 
-        return self._states.copy(), rewards, absorbings, step_infos
+        return self._states.copy(), rewards.copy(), absorbings.copy(), step_infos.copy()
 
     def render_all(self, env_mask, record=False):
         for i, remote in enumerate(self._remotes):