fix Issue 2039

2019-01-28 16:24:44 -08:00
parent 0818112ede
commit 63683e8f02
1 changed files with 2 additions and 0 deletions
--- a/examples/pybullet/gym/pybullet_envs/robot_bases.py
+++ b/examples/pybullet/gym/pybullet_envs/robot_bases.py
@@ -235,6 +235,8 @@ class BodyPart:
 			(x, y, z), (a, b, c, d), _, _, _, _ = self._p.getLinkState(body_id, link_id)
 		return np.array([x, y, z, a, b, c, d])

+	def get_position(self): return self.current_position()
+
 	def get_pose(self):
 		return self.state_fields_of_pose_of(self.bodies[self.bodyIndex], self.bodyPartIndex)