Removed hack again...
This commit is contained in:
parent
f86c6c736f
commit
f421dc2ab5
@ -236,8 +236,6 @@ class GaussianRolloutCollectorAuxclass():
|
|||||||
if callback.on_step() is False:
|
if callback.on_step() is False:
|
||||||
return False
|
return False
|
||||||
|
|
||||||
if isinstance(env, DummyVecEnv):
|
|
||||||
self._update_info_buffer(infos[0])
|
|
||||||
self._update_info_buffer(infos)
|
self._update_info_buffer(infos)
|
||||||
n_steps += 1
|
n_steps += 1
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user