admin/lerobot
1
0
Fork 0
mirror of https://github.com/huggingface/lerobot.git synced 2026-05-18 02:00:03 +00:00
Code Issues Packages Projects Releases Wiki Activity

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse Source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot]
2025-08-28 07:53:10 +00:00
parent 2b03dec01f
commit 213ffe02cf
1 changed files with 1 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/artifacts/datasets/save_dataset_to_safetensors.py
+1 -5
View File
@@ -53,11 +53,7 @@ def save_dataset_to_safetensors(output_dir, repo_id="lerobot/pusht"):
# save 2 frames at the middle of first episode
i = int(
(
dataset.meta.episodes["dataset_to_index"][0]
- dataset.meta.episodes["dataset_from_index"][0]
)
/ 2
(dataset.meta.episodes["dataset_to_index"][0] - dataset.meta.episodes["dataset_from_index"][0]) / 2
)
save_file(dataset[i], repo_dir / f"frame_{i}.safetensors")
save_file(dataset[i + 1], repo_dir / f"frame_{i + 1}.safetensors")