Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
fix get_camera_dirs to use full content name and validate prefix
Use all parts after "observation." joined with "_" for content_name
(e.g. observation.images.top -> images_top) instead of only the last
part. Also replace assert with FastLabelInvalidException for
non-observation camera directories.

Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
  • Loading branch information
rikunosuke and claude committed Mar 19, 2026
commit 6cddb6558a8b511ec7441da5c568ae393b7a2cb9
9 changes: 7 additions & 2 deletions fastlabel/lerobot/common.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def format_episode_name(episode_index: int) -> str:
def get_camera_dirs(lerobot_data_path: Path) -> list:
"""Get camera directories and their content names.
Returns [(camera_dir, content_name), ...].
e.g. observation.images.top -> content_name = "top"
e.g. observation.images.top -> content_name = "images_top"
"""
videos_dir = lerobot_data_path / "videos"
if not videos_dir.exists():
Expand All @@ -63,6 +63,11 @@ def get_camera_dirs(lerobot_data_path: Path) -> list:
if not obs_dir.is_dir():
continue
parts = obs_dir.name.split(".")
content_name = parts[-1] if len(parts) > 1 else obs_dir.name
if parts[0] != "observation":
raise FastLabelInvalidException(
f"Unexpected camera dir name: {obs_dir.name}"
)

content_name = "_".join(parts[1:])
results.append((obs_dir, content_name))
return results
Loading