hangX/lerobot
1
0
Fork 0
forked from tangger/lerobot
Code Pull Requests Activity

Nits

Browse Source
This commit is contained in:
Simon Alibert
2024-10-23 20:55:54 +02:00
parent 1aba80d93f
commit 0098bd264e
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -389,7 +389,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
@property
def features(self) -> datasets.Features:
"""Shapes for the different features."""
"""Features of the hf_dataset."""
if self.hf_dataset is not None:
return self.hf_dataset.features
elif self.episode_buffer is None:
@@ -664,7 +664,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
self.episode_dicts.append(episode_dict)
append_jsonl(episode_dict, self.root / EPISODES_PATH)
def delete_episode(self) -> None:
def clear_episode_buffer(self) -> None:
episode_index = self.episode_buffer["episode_index"]
if self.image_writer is not None:
for cam_key in self.camera_keys: