diff --git a/myosuite/envs/myo/myochallenge/bimanual_v0.py b/myosuite/envs/myo/myochallenge/bimanual_v0.py index bf5f41df..7bf225ee 100644 --- a/myosuite/envs/myo/myochallenge/bimanual_v0.py +++ b/myosuite/envs/myo/myochallenge/bimanual_v0.py @@ -78,6 +78,7 @@ def _setup(self, self.start_shifts = start_shifts self.goal_shifts = goal_shifts + self.PILLAR_HEIGHT = 1.09 self.id_info = IdInfo(self.sim.model) @@ -246,7 +247,7 @@ def get_reward_dict(self, obs_dict): obj_pos = obs_dict["obj_pos"][0][0] if obs_dict['obj_pos'].ndim == 3 else obs_dict['obj_pos'] palm_pos = obs_dict["palm_pos"][0][0] if obs_dict["palm_pos"].ndim == 3 else obs_dict["palm_pos"] goal_pos = obs_dict["goal_pos"][0][0] if obs_dict["goal_pos"].ndim == 3 else obs_dict["goal_pos"] - goal_pos = np.concatenate((goal_pos[:2], np.array([1.09]))) + goal_pos = np.concatenate((goal_pos[:2], np.array([self.PILLAR_HEIGHT]))) lift_height = np.linalg.norm(np.array([[[obj_pos[-1], palm_pos[-1]]]]) - np.array([[[self.init_obj_z, self.init_palm_z]]]), axis=-1)