updates to point env randgoal - make it 1d

cbfinn · Jan 19, 2017 · 8126059 · 8126059
1 parent c94b235
commit 8126059
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/examples/point_env_randgoal.py b/examples/point_env_randgoal.py
@@ -21,10 +21,13 @@ def reset(self, reset_args=None):
  goal = reset_args
  if goal is not None:
  self._goal = goal
- elif self._goal is None:
+ #elif self._goal is None:
+ else:
  # Only set a new goal if this env hasn't had one defined before.
- self._goal = np.random.uniform(-1, 1, size=(2,))
- self._state = (0, 0) #np.random.uniform(-1, 1, size=(2,))
+ self._goal = np.random.uniform(0, 1, size=(2,))
+ self._goal[1] = 0
+
+ self._state = (0, 0)
  observation = np.copy(self._state)
  return observation