From 1329954dba89d0e2e3fbd7834e74574fc5561bea Mon Sep 17 00:00:00 2001 From: Jade Choghari Date: Mon, 24 Feb 2025 14:23:24 +0100 Subject: [PATCH] merge changes --- lerobot/common/datasets/lerobot_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lerobot/common/datasets/lerobot_dataset.py b/lerobot/common/datasets/lerobot_dataset.py index 0986ca54..b7e7bca9 100644 --- a/lerobot/common/datasets/lerobot_dataset.py +++ b/lerobot/common/datasets/lerobot_dataset.py @@ -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():