Skip to content

Commit

Permalink
format
Browse files Browse the repository at this point in the history
  • Loading branch information
YilingQiao committed Dec 19, 2024
1 parent 1c9ea4d commit 2985aa9
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion examples/locomotion/go2_env.py
Original file line number Diff line number Diff line change
Expand Up @@ -154,7 +154,7 @@ def step(self, actions):
time_out_idx = (self.episode_length_buf > self.max_episode_length).nonzero(as_tuple=False).flatten()
self.extras["time_outs"] = torch.zeros_like(self.reset_buf, device=self.device, dtype=gs.tc_float)
self.extras["time_outs"][time_out_idx] = 1.0

self.reset_idx(self.reset_buf.nonzero(as_tuple=False).flatten())

# compute reward
Expand Down

0 comments on commit 2985aa9

Please sign in to comment.