diff --git a/src/lerobot/scripts/lerobot_edit_dataset.py b/src/lerobot/scripts/lerobot_edit_dataset.py index 7c58389ac..e835b1de6 100644 --- a/src/lerobot/scripts/lerobot_edit_dataset.py +++ b/src/lerobot/scripts/lerobot_edit_dataset.py @@ -379,9 +379,9 @@ def encode_episode_videos( img_keys = [key for key in hf_dataset.features if key.startswith(OBS_IMAGE)] video_metadata = {} - fps = dataset.fps + fps = int(dataset.fps) # Convert to int for PyAV compatibility 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: # Save images temporarily