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

fix linter 5

Browse Source
This commit is contained in:
Martino Russi
2025-11-27 13:15:31 +01:00
parent d8e69c637a
commit 998d108424
1 changed files with 3 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
examples/unitree_g1/gr00t_locomotion.py
+3 -1
View File
@@ -218,7 +218,9 @@ class GrootLocomotionController:
cmd_magnitude = np.linalg.norm(self.locomotion_cmd) cmd_magnitude = np.linalg.norm(self.locomotion_cmd)
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 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 # run policy inference
ort_inputs = {selected_policy.get_inputs()[0].name: obs_tensor.cpu().numpy()} ort_inputs = {selected_policy.get_inputs()[0].name: obs_tensor.cpu().numpy()}