tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Apply suggestions from code review 3

Browse Source 
Co-authored-by: Simon Alibert <75076266+aliberts@users.noreply.github.com>
This commit is contained in:
Steven Palma
2025-05-20 19:10:06 +02:00
committed by GitHub
parent 33b8559060
commit a36536dc07
1 changed files with 6 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
lerobot/common/robots/config.py
View File

@@ -27,14 +27,12 @@ class RobotConfig(draccus.ChoiceRegistry, abc.ABC):
calibration_dir: Path | None = None calibration_dir: Path | None = None
def __post_init__(self): def __post_init__(self):
if hasattr(self, "cameras"): if hasattr(self, "cameras") and self.cameras:
cameras = self.cameras for name, config in self.cameras.items():
if cameras: for attr in ["width", "height", "fps"]:
for cam_name, cam_config in cameras.items(): if getattr(config, attr) is None:
for attr in ["width", "height", "fps"]: raise ValueError(
if getattr(cam_config, attr) is None: f"Specifying '{attr}' is required for the camera to be used in a robot"
raise ValueError(
f"Camera config for '{cam_name}' has None value for required attribute '{attr}'"
) )
@property @property