admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-26 05:59:52 +00:00
Code Issues Packages Projects Releases Wiki Activity

more fixes

Browse Source
This commit is contained in:
Jade Choghari
2025-12-05 20:37:27 +00:00
parent ba97f64afd
commit 8b7c46c5f7
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/scripts/lerobot_edit_dataset.py
+2 -2
View File
@@ -379,9 +379,9 @@ def encode_episode_videos(
img_keys = [key for key in hf_dataset.features if key.startswith(OBS_IMAGE)] img_keys = [key for key in hf_dataset.features if key.startswith(OBS_IMAGE)]
video_metadata = {} video_metadata = {}
fps = dataset.fps fps = int(dataset.fps) # Convert to int for PyAV compatibility
episode_length = dataset.meta.episodes["length"][episode_index] episode_length = dataset.meta.episodes["length"][episode_index]
episode_duration = episode_length / fps episode_duration = episode_length / dataset.fps # Use original fps for duration calculation
for img_key in img_keys: for img_key in img_keys:
# Save images temporarily # Save images temporarily