From e4b17e03f68570e517400b09e1ee9c32780c9ff5 Mon Sep 17 00:00:00 2001 From: Eric Liang Date: Fri, 13 Apr 2018 00:57:00 -0700 Subject: [PATCH] updates (#1896) --- examples/custom_env/custom_env.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/examples/custom_env/custom_env.py b/examples/custom_env/custom_env.py index 87aa757db..b5a3240ea 100644 --- a/examples/custom_env/custom_env.py +++ b/examples/custom_env/custom_env.py @@ -27,11 +27,11 @@ class SimpleCorridor(gym.Env): 0.0, self.end_pos, shape=(1,), dtype=np.float32) self._spec = EnvSpec("SimpleCorridor-{}-v0".format(self.end_pos)) - def _reset(self): + def reset(self): self.cur_pos = 0 return [self.cur_pos] - def _step(self, action): + def step(self, action): assert action in [0, 1] if action == 0 and self.cur_pos > 0: self.cur_pos -= 1