The reward_spec and discount_spec methods are not in Wrapper. As such when wrapping an environment with a reward/discount spec that is not the default (shape=()), the reward and discount specs are set to the default unless they are specified again in the wrapper.
The
reward_spec
anddiscount_spec
methods are not inWrapper
. As such when wrapping an environment with a reward/discount spec that is not the default (shape=()
), the reward and discount specs are set to the default unless they are specified again in the wrapper.For example