fix distance reward calculation

2019-07-01 10:26:56 +02:00
parent 9addf64ee9
commit 88120527b6
1 changed files with 1 additions and 1 deletions
--- a/examples/pybullet/gym/pybullet_envs/deep_mimic/env/pybullet_deep_mimic_env.py
+++ b/examples/pybullet/gym/pybullet_envs/deep_mimic/env/pybullet_deep_mimic_env.py
@@ -409,7 +409,7 @@ class PyBulletDeepMimicEnv(Env):
    if self.goal.is_hit:
      return 1
    else:
-      goalPos = self.goal.goal_data
+      goalPos = self.goal.world_pos
      distanceSquared = sum([(x - y)**2 for (x, y) in zip(goalPos, linkPos)])
      return math.exp(-4*distanceSquared)