Removed hack again...

2023-01-26 18:18:38 +01:00 · 2023-01-26 18:18:38 +01:00 · f421dc2ab5
commit f421dc2ab5
parent f86c6c736f
1 changed files with 0 additions and 2 deletions
--- a/metastable_baselines/misc/rollout_buffer.py
+++ b/metastable_baselines/misc/rollout_buffer.py
@ -236,8 +236,6 @@ class GaussianRolloutCollectorAuxclass():
            if callback.on_step() is False:
                return False

-            if isinstance(env, DummyVecEnv):
-                self._update_info_buffer(infos[0])
            self._update_info_buffer(infos)
            n_steps += 1