diff --git a/normalized_actions.py b/normalized_actions.py index d1814f5..a0e64c7 100644 --- a/normalized_actions.py +++ b/normalized_actions.py @@ -3,7 +3,7 @@ import gym class NormalizedActions(gym.ActionWrapper): - def _action(self, action): + def action(self, action): action = (action + 1) / 2 # [-1, 1] => [0, 1] action *= (self.action_space.high - self.action_space.low) action += self.action_space.low