hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Fix dict

Browse Source
This commit is contained in:
Simon Alibert
2025-03-24 12:16:54 +01:00
parent 39d8f45810
commit 9832daf08d
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -263,8 +263,8 @@ class MotorsBus(abc.ABC):
self.calibration = None
self._id_to_model = {m.id: m.model for m in self.motors.values()}
self._id_to_name = {m.id: name for name, m in self.motors.items()}
self._id_to_model_dict = {m.id: m.model for m in self.motors.values()}
self._id_to_name_dict = {m.id: name for name, m in self.motors.items()}
def __len__(self):
return len(self.motors)
@@ -298,10 +298,10 @@ class MotorsBus(abc.ABC):
return [m.id for m in self.motors.values()]
def _id_to_model(self, motor_id: int) -> str:
return self._id_to_model[motor_id]
return self._id_to_model_dict[motor_id]
def _id_to_name(self, motor_id: int) -> str:
return self._id_to_name[motor_id]
return self._id_to_name_dict[motor_id]
def _get_motor_id(self, motor: NameOrID) -> int:
if isinstance(motor, str):