saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-07-29 20:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fixes update_locals() in collect_rollouts() of OnPolicyAlgorithm (#1660)

Browse source 
* calls update_locals() before on_rollout_end()

* update changelog
This commit is contained in:
PatrickHelm 2023-08-30 17:02:41 +02:00 committed by GitHub
parent c99d65c664
commit 84163b468c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 3 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
docs/misc/changelog.rst
View file

@ -22,6 +22,7 @@ Bug Fixes:
^^^^^^^^^^
- Moves VectorizedActionNoise into ``_setup_learn()`` in OffPolicyAlgorithm (@PatrickHelm)
- Prevents out of bound error on Windows if no seed is passed (@PatrickHelm)
- Calls ``callback.update_locals()`` before ``callback.on_rollout_end()`` in OnPolicyAlgorithm (@PatrickHelm)
Deprecations:
^^^^^^^^^^^^^

2
stable_baselines3/common/on_policy_algorithm.py
View file

@ -221,6 +221,8 @@ class OnPolicyAlgorithm(BaseAlgorithm):
rollout_buffer.compute_returns_and_advantage(last_values=values, dones=dones)
callback.update_locals(locals())
callback.on_rollout_end()
return True