admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-18 02:00:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix linter

Browse Source
This commit is contained in:
Martino Russi
2025-11-27 10:43:15 +01:00
parent 36ed02adfa
commit 9a90b7dcb2
3 changed files with 10 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/unitree_g1/gr00t_locomotion.py
+1 -6
View File
@@ -218,12 +218,7 @@ class GrootLocomotionController:
cmd_magnitude = np.linalg.norm(self.locomotion_cmd)
if cmd_magnitude < 0.05:
# balance/standing policy for small commands
selected_policy = self.policy_balance
else:
# walking policy for movement commands
selected_policy = self.policy_walk
selected_policy = self.policy_balance if cmd_magnitude < 0.05 else self.policy_walk #balance/standing policy for small commands, walking policy for movement commands
# run policy inference
ort_inputs = {selected_policy.get_inputs()[0].name: obs_tensor.cpu().numpy()}