Source code for lagom.envs.wrappers.sign_clip_reward

import numpy as np

from gym import RewardWrapper


class SignClipReward(RewardWrapper):
    r""""Bin reward to {-1, 0, +1} by its sign. """   
    def reward(self, reward):
        return np.sign(reward)