evorl.envs.wrappers.reward_wrapper

Module Contents

Classes

RewardScaleWrapper	Scale the reward by a factor.
SparseRewardWrapper	Convert dense reward to sparse reward.

API

class evorl.envs.wrappers.reward_wrapper.RewardScaleWrapper(env: evorl.envs.env.Env, reward_scale: float)

Bases: evorl.envs.wrappers.wrapper.Wrapper

Scale the reward by a factor.

Usage:

• Use EpisodeWrapper(RewardScaleWrapper(env)) to get the scaled info.episode_return.

• Use RewardScaleWrapper(EpisodeWrapper(env)) to get the original info.episode_return.

reset(key: chex.PRNGKey) → evorl.envs.env.EnvState

step(state: evorl.envs.env.EnvState, action: evorl.types.Action) → evorl.envs.env.EnvState

class evorl.envs.wrappers.reward_wrapper.SparseRewardWrapper(env: evorl.envs.env.Env, sparse_length: int)

Bases: evorl.envs.wrappers.wrapper.Wrapper

Convert dense reward to sparse reward.

The dense rewards become: 0, 0, …, sum(rewards), 0, 0, …, sum(rewards)

reset(key: chex.PRNGKey) → evorl.envs.env.EnvState

step(state: evorl.envs.env.EnvState, action: evorl.types.Action) → evorl.envs.env.EnvState