From 9dca233d7eb4d2e7842256a0068c36904ae5816f Mon Sep 17 00:00:00 2001 From: Simon Alibert Date: Wed, 23 Oct 2024 00:27:14 +0200 Subject: [PATCH] Fix episode chunk --- lerobot/common/datasets/lerobot_dataset.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/lerobot/common/datasets/lerobot_dataset.py b/lerobot/common/datasets/lerobot_dataset.py index acde3b922..014d27831 100644 --- a/lerobot/common/datasets/lerobot_dataset.py +++ b/lerobot/common/datasets/lerobot_dataset.py @@ -305,10 +305,7 @@ def get_video_file_path(self, ep_index: int, vid_key: str) -> Path: return Path(fpath) def get_episode_chunk(self, ep_index: int) -> int: - ep_chunk = ep_index // self.chunks_size - if ep_index > 0 and ep_index % self.chunks_size == 0: - ep_chunk -= 1 - return ep_chunk + return ep_index // self.chunks_size @property def data_path(self) -> str: