def __init__(self, env: gym.Env, scale):
    gym.RewardWrapper.__init__(self, env)
    self.scale = scale
def __init__(self, env: gym.Env, scale):
    gym.RewardWrapper.__init__(self, env)
    self.scale = scale