saymrwulf/stable-baselines3
1
0
Fork 0
mirror of https://github.com/saymrwulf/stable-baselines3.git synced 2026-09-17 22:30:59 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix DictReplayBuffer.next_observations type (#1013)

Browse source 
* Fix DictReplayBuffer.next_observations type

* Update changelog

Co-authored-by: Antonin RAFFIN <antonin.raffin@ensta.org>
This commit is contained in:
Quentin Gallouédec 2022-08-16 10:53:22 +02:00 committed by GitHub
parent c4f54fcf04
commit a30d36002b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
docs/misc/changelog.rst
View file

@ -26,6 +26,7 @@ Deprecations:
Others: Others:
^^^^^^^ ^^^^^^^
- Fixed ``DictReplayBuffer.next_observations`` typing (@qgallouedec)
Documentation: Documentation:
^^^^^^^^^^^^^^ ^^^^^^^^^^^^^^

2
stable_baselines3/common/type_aliases.py
View file

@ -50,7 +50,7 @@ class ReplayBufferSamples(NamedTuple):
class DictReplayBufferSamples(ReplayBufferSamples): class DictReplayBufferSamples(ReplayBufferSamples):
observations: TensorDict observations: TensorDict
actions: th.Tensor actions: th.Tensor
next_observations: th.Tensor next_observations: TensorDict
dones: th.Tensor dones: th.Tensor
rewards: th.Tensor rewards: th.Tensor