admin/any4lerobot
1
0
Fork 0
mirror of https://github.com/Tavish9/any4lerobot.git synced 2026-06-23 14:27:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply suggestions from code review

Browse Source 
Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
This commit is contained in:
Qizhi Chen
2026-06-21 22:22:17 -07:00
committed by GitHub
parent 8afce7fa23
commit 43a76d30ce
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
agibot2lerobot/agibot_utils/agibot_utils.py
+4 -3
View File
@@ -108,9 +108,10 @@ def load_local_dataset(
if save_depth:
depth_imgs = load_depths(ob_dir / "depth", "head_depth")
assert num_frames == len(depth_imgs), (
"Number of images and states are not equal"
)
if num_frames != len(depth_imgs):
raise ValueError(
f"Number of images ({len(depth_imgs)}) and states ({num_frames}) are not equal"
)
state_key_prefix_len = len("observation.states.")
action_key_prefix_len = len("actions.")