From 2985aa9c63b4162cea10a52c98161e1270477746 Mon Sep 17 00:00:00 2001 From: YilingQiao Date: Thu, 19 Dec 2024 09:11:47 -0500 Subject: [PATCH] format --- examples/locomotion/go2_env.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/locomotion/go2_env.py b/examples/locomotion/go2_env.py index 4368b8ab..f3d8f17f 100644 --- a/examples/locomotion/go2_env.py +++ b/examples/locomotion/go2_env.py @@ -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