From f421dc2ab5e4bdb3ba21416f2bc258aa3cb7ac7c Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Thu, 26 Jan 2023 18:18:38 +0100
Subject: [PATCH] Removed hack again...

---
 metastable_baselines/misc/rollout_buffer.py | 2 --
 1 file changed, 2 deletions(-)

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