Closed ernestum closed 11 months ago
This adds the rollout_buffer_class to TRPO in sync with https://github.com/DLR-RM/stable-baselines3/pull/1720. I decided not to add it to MaskeablePPO and RecurrentPPO since they require special rollout buffers.
make format
make check-codestyle
make lint
make pytest
make type
Note: we are using a maximum length of 127 characters per line
Fyi, I already did a pre-release https://pypi.org/project/stable-baselines3/2.2.0a9/, and you would need it to build this SB3 contrib version ;)
Oh nice thanks!
Description
This adds the rollout_buffer_class to TRPO in sync with https://github.com/DLR-RM/stable-baselines3/pull/1720. I decided not to add it to MaskeablePPO and RecurrentPPO since they require special rollout buffers.
Context
Types of changes
Checklist:
make format
(required)make check-codestyle
andmake lint
(required)make pytest
andmake type
both pass. (required)Note: we are using a maximum length of 127 characters per line