diff --git a/envs.py b/envs.py index 0724432..4339369 100644 --- a/envs.py +++ b/envs.py @@ -31,7 +31,7 @@ def __init__(self, env=None): super(AtariRescale42x42, self).__init__(env) self.observation_space = Box(0.0, 1.0, [1, 42, 42]) - def _observation(self, observation): + def observation(self, observation): return _process_frame42(observation) @@ -43,7 +43,7 @@ def __init__(self, env=None): self.alpha = 0.9999 self.num_steps = 0 - def _observation(self, observation): + def observation(self, observation): self.num_steps += 1 self.state_mean = self.state_mean * self.alpha + \ observation.mean() * (1 - self.alpha)