RewardScalingWrapper

class maze.core.wrappers.reward_scaling_wrapper.RewardScalingWrapper(*args, **kwds)

Scales original step reward by a multiplicative scaling factor.

Parameters

• env – The underlying environment.

• scale – Multiplicative reward scaling factor.

reward(reward: float) → float

Scales the original reward.

Parameters

reward – The original reward.

Returns

The scaled reward.