Hack is back

2023-01-26 18:15:06 +01:00 · 2023-01-26 18:15:06 +01:00 · f86c6c736f
commit f86c6c736f
parent 20646f07f3
1 changed files with 2 additions and 0 deletions
--- a/metastable_baselines/misc/rollout_buffer.py
+++ b/metastable_baselines/misc/rollout_buffer.py
@ -236,6 +236,8 @@ 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