From ca08cefe9cc7a01d60c97a318a5a60bfa5a0d01d Mon Sep 17 00:00:00 2001 From: Brian Timar Date: Wed, 25 Sep 2019 14:04:56 -0400 Subject: [PATCH] rename observation method --- envs.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) 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)