wangzixiang
/
lerobot
mirror of https://github.com/huggingface/lerobot.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

merge changes

This commit is contained in:
Jade Choghari 2025-02-24 14:23:24 +01:00
parent dc4e94fc65
commit 1329954dba
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -695,7 +695,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
# data logic
if self.delta_indices is not None:
current_ep_idx = self.episodes.index(ep_idx) if self.episodes is not None else ep_idx
query_indices, padding = self._get_query_indices(idx, current_ep_idx)
query_indices, padding = self._get_query_indices(idx, current_ep_idx) #
query_result = self._query_hf_dataset(query_indices)
item = {**item, **padding}
for key, val in query_result.items():