diff --git a/docs/misc/changelog.rst b/docs/misc/changelog.rst index 1d7a288..41e40ba 100644 --- a/docs/misc/changelog.rst +++ b/docs/misc/changelog.rst @@ -28,6 +28,7 @@ Others: ^^^^^^^ - Fixed flake8 config - Fixed ``sb3_contrib/common/utils.py`` type hint +- Fixed ``sb3_contrib/common/recurrent/type_aliases.py`` type hint Release 1.6.2 (2022-10-10) -------------------------- diff --git a/sb3_contrib/common/recurrent/type_aliases.py b/sb3_contrib/common/recurrent/type_aliases.py index 1ae9a08..21ac0e0 100644 --- a/sb3_contrib/common/recurrent/type_aliases.py +++ b/sb3_contrib/common/recurrent/type_aliases.py @@ -21,7 +21,7 @@ class RecurrentRolloutBufferSamples(NamedTuple): mask: th.Tensor -class RecurrentDictRolloutBufferSamples(RecurrentRolloutBufferSamples): +class RecurrentDictRolloutBufferSamples(NamedTuple): observations: TensorDict actions: th.Tensor old_values: th.Tensor diff --git a/setup.cfg b/setup.cfg index dad73ae..4004a6b 100644 --- a/setup.cfg +++ b/setup.cfg @@ -29,7 +29,6 @@ exclude = (?x)( | sb3_contrib/ars/policies.py$ | sb3_contrib/qrdqn/qrdqn.py$ | sb3_contrib/qrdqn/policies.py$ - | sb3_contrib/common/recurrent/type_aliases.py$ | sb3_contrib/common/recurrent/policies.py$ | sb3_contrib/common/recurrent/buffers.py$ | sb3_contrib/common/maskable/distributions.py$