Skip to content

Commit 7157863

Browse files
dense rewards for button press fixed (#548)
1 parent 5396509 commit 7157863

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

metaworld/envs/sawyer_button_press_topdown_wall_v3.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -139,7 +139,7 @@ def compute_reward(
139139
tcp_to_obj_init = float(np.linalg.norm(obj - self.init_tcp))
140140
obj_to_target = abs(self._target_pos[2] - obj[2])
141141

142-
tcp_closed = 1 - obs[3]
142+
tcp_closed = max(obs[3], 0.0)
143143
near_button = reward_utils.tolerance(
144144
tcp_to_obj,
145145
bounds=(0, 0.01),

0 commit comments

Comments
 (0)