Skip to content

Commit

Permalink
temp
Browse files Browse the repository at this point in the history
  • Loading branch information
arth-shukla committed Jan 22, 2024
1 parent 3ea9c06 commit ecb296f
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions mani_skill2/envs/sapien_env.py
Original file line number Diff line number Diff line change
Expand Up @@ -624,6 +624,7 @@ def step(self, action: Union[None, np.ndarray, Dict]):
info = self.get_info(obs=obs)
reward = self.get_reward(obs=obs, action=action, info=info)
terminated = info["success"]
print(info, terminated)
if self.num_envs == 1:
terminated = terminated[0]
reward = reward[0]
Expand Down

0 comments on commit ecb296f

Please sign in to comment.