馃殌 Feature
Add next_observations and dones fields to the RolloutBuffer and the DictRolloutBuffer classes, similar to how it is done in the ReplayBuffer class.
Motivation
Currently, on-policy algorithms don't store the next observations and dones fields in their buffer in the get_rollouts method. This is because these fields are not required by any of the algorithms in stable-baselines3. However, these fields are required to be stored in the buffer to implement the original variant of the AIRL algorithm in imitation.
Pitch
No response
Alternatives
No response
Additional context
No response
Checklist
馃殌 Feature
Add
next_observationsanddonesfields to theRolloutBufferand theDictRolloutBufferclasses, similar to how it is done in theReplayBufferclass.Motivation
Currently, on-policy algorithms don't store the next observations and dones fields in their buffer in the
get_rolloutsmethod. This is because these fields are not required by any of the algorithms in stable-baselines3. However, these fields are required to be stored in the buffer to implement the original variant of the AIRL algorithm in imitation.Pitch
No response
Alternatives
No response
Additional context
No response
Checklist