diff --git a/metastable_baselines/misc/rollout_buffer.py b/metastable_baselines/misc/rollout_buffer.py index 1fea969..846d60a 100644 --- 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