diff --git a/gym_puddle/envs/puddle_env.py b/gym_puddle/envs/puddle_env.py index 2752ca4..832c178 100644 --- a/gym_puddle/envs/puddle_env.py +++ b/gym_puddle/envs/puddle_env.py @@ -25,14 +25,14 @@ def __init__(self, start=[0.2, 0.4], goal=[1.0, 1.0], goal_threshold=0.1, for i in range(4): self.actions[i][i//2] = thrust * (i%2 * 2 - 1) - self._seed() + self.seed() self.viewer = None - def _seed(self, seed=None): + def seed(self, seed=None): self.np_random, seed = seeding.np_random(seed) return [seed] - def _step(self, action): + def step(self, action): assert self.action_space.contains(action), "%r (%s) invalid"%(action, type(action)) self.pos += self.actions[action] + self.np_random.uniform(low=-self.noise, high=self.noise, size=(2,)) @@ -55,14 +55,14 @@ def _get_reward(self, pos): def _gaussian1d(self, p, mu, sig): return np.exp(-((p - mu)**2)/(2.*sig**2)) / (sig*np.sqrt(2.*np.pi)) - def _reset(self): + def reset(self): if self.start is None: self.pos = self.observation_space.sample() else: self.pos = copy.copy(self.start) return self.pos - def _render(self, mode='human', close=False): + def render(self, mode='human', close=False): if close: if self.viewer is not None: self.viewer.close()