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

feat(parallel encoding): making parallel encoding the default choice over all platforms (#2525)

Browse Source
This commit is contained in:
Caroline Pascal
2025-11-26 14:57:34 +01:00
committed by GitHub
parent 17581a9449
commit 581dd45eae
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/lerobot_dataset.py
+1 -2
View File
@@ -16,7 +16,6 @@
import concurrent.futures import concurrent.futures
import contextlib import contextlib
import logging import logging
import platform
import shutil import shutil
import tempfile import tempfile
from collections.abc import Callable from collections.abc import Callable
@@ -1149,7 +1148,7 @@ class LeRobotDataset(torch.utils.data.Dataset):
def save_episode( def save_episode(
self, self,
episode_data: dict | None = None, episode_data: dict | None = None,
parallel_encoding: bool = platform.system() == "Linux", parallel_encoding: bool = True,
) -> None: ) -> None:
""" """
This will save to disk the current episode in self.episode_buffer. This will save to disk the current episode in self.episode_buffer.