tangger/lerobot
1
0
Fork 1
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

Optimized episode cache verification (#2166)

Browse Source 
Signed-off-by: Antoine <antoine.dandigne@gmail.com>
Co-authored-by: Michel Aractingi <michel.aractingi@huggingface.co>
This commit is contained in:
Antoine
2025-10-17 15:18:21 +02:00
committed by GitHub
parent ed49c9935a
commit a51682b266
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
src/lerobot/datasets/lerobot_dataset.py
View File

@@ -842,7 +842,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
# Get available episode indices from cached dataset # Get available episode indices from cached dataset
available_episodes = { available_episodes = {
ep_idx.item() if isinstance(ep_idx, torch.Tensor) else ep_idx ep_idx.item() if isinstance(ep_idx, torch.Tensor) else ep_idx
for ep_idx in self.hf_dataset["episode_index"] for ep_idx in self.hf_dataset.unique("episode_index")
} }
# Determine requested episodes # Determine requested episodes