admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-07-22 17:32:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

lerobot dataset fix

Browse Source
This commit is contained in:
Michel Aractingi
2025-12-10 16:42:16 +01:00
parent 4bcd14b8de
commit 08d2ed8015
1 changed files with 0 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/lerobot_dataset.py
-5
View File
@@ -1402,7 +1402,6 @@ class LeRobotDataset(torch.utils.data.Dataset):
return metadata return metadata
def _save_episode_video(self, video_key: str, episode_index: int, video_path: str | Path | None = None) -> dict:
def _save_episode_video( def _save_episode_video(
self, self,
video_key: str, video_key: str,
@@ -1410,10 +1409,6 @@ class LeRobotDataset(torch.utils.data.Dataset):
temp_path: Path | None = None, temp_path: Path | None = None,
) -> dict: ) -> dict:
# Encode episode frames into a temporary video # Encode episode frames into a temporary video
if video_path is None:
ep_path = self._encode_temporary_episode_video(video_key, episode_index)
else:
ep_path = video_path
if temp_path is None: if temp_path is None:
ep_path = self._encode_temporary_episode_video(video_key, episode_index) ep_path = self._encode_temporary_episode_video(video_key, episode_index)
else: else: