From 63683e8f02f68e277193ee4309ff9ea06f2a110a Mon Sep 17 00:00:00 2001 From: Erwin Coumans Date: Mon, 28 Jan 2019 16:24:44 -0800 Subject: [PATCH] fix Issue 2039 --- examples/pybullet/gym/pybullet_envs/robot_bases.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/examples/pybullet/gym/pybullet_envs/robot_bases.py b/examples/pybullet/gym/pybullet_envs/robot_bases.py index 18fd3defe..36e5b8c76 100644 --- 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)