wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks 

for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2025-04-08 08:48:17 +00:00
parent 8924ccbbab
commit 782dff1163
2 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
lerobot/common/motors/motors_bus.py
View File

@ -353,7 +353,7 @@ class MotorsBus(abc.ABC):
def _get_ids_values_dict(self, values: Value | dict[str, Value] | None) -> list[str]: def _get_ids_values_dict(self, values: Value | dict[str, Value] | None) -> list[str]:
if isinstance(values, (int, float)): if isinstance(values, (int, float)):
return {id_: values for id_ in self.ids} return dict.fromkeys(self.ids, values)
elif isinstance(values, dict): elif isinstance(values, dict):
return {self.motors[motor].id: val for motor, val in values.items()} return {self.motors[motor].id: val for motor, val in values.items()}
else: else:
@ -499,7 +499,7 @@ class MotorsBus(abc.ABC):
try: try:
drive_modes = self.sync_read("Drive_Mode", normalize=False) drive_modes = self.sync_read("Drive_Mode", normalize=False)
except KeyError: except KeyError:
drive_modes = {name: 0 for name in self.names} drive_modes = dict.fromkeys(self.names, 0)
calibration = {} calibration = {}
for name, motor in self.motors.items(): for name, motor in self.motors.items():

2
lerobot/common/robots/utils.py
View File

@ -85,7 +85,7 @@ def ensure_safe_goal_position(
"""Caps relative action target magnitude for safety.""" """Caps relative action target magnitude for safety."""
if isinstance(max_relative_target, float): if isinstance(max_relative_target, float):
diff_cap = {key: max_relative_target for key in goal_present_pos} diff_cap = dict.fromkeys(goal_present_pos, max_relative_target)
elif isinstance(max_relative_target, dict): elif isinstance(max_relative_target, dict):
if not set(goal_present_pos) == set(max_relative_target): if not set(goal_present_pos) == set(max_relative_target):
raise ValueError("max_relative_target keys must match those of goal_present_pos.") raise ValueError("max_relative_target keys must match those of goal_present_pos.")