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

revert(datasets): use main package exports

Browse Source
This commit is contained in:
Pepijn
2026-07-15 15:12:17 +02:00
parent ca5be5b482
commit d3ad24d9dd
1 changed files with 1 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/lerobot/datasets/__init__.py
+1 -19
View File
@@ -35,6 +35,7 @@ from .dataset_tools import (
remove_feature, remove_feature,
split_dataset, split_dataset,
) )
from .factory import make_dataset, make_train_eval_datasets, resolve_delta_timestamps
from .image_writer import safe_stop_image_writer from .image_writer import safe_stop_image_writer
from .io_utils import load_episodes, write_stats from .io_utils import load_episodes, write_stats
from .language import ( from .language import (
@@ -54,25 +55,6 @@ from .streaming_dataset import StreamingLeRobotDataset
from .utils import DEFAULT_EPISODES_PATH, create_lerobot_dataset_card from .utils import DEFAULT_EPISODES_PATH, create_lerobot_dataset_card
from .video_utils import VideoEncodingManager from .video_utils import VideoEncodingManager
def make_dataset(*args, **kwargs):
from .factory import make_dataset as _make_dataset
return _make_dataset(*args, **kwargs)
def make_train_eval_datasets(*args, **kwargs):
from .factory import make_train_eval_datasets as _make_train_eval_datasets
return _make_train_eval_datasets(*args, **kwargs)
def resolve_delta_timestamps(*args, **kwargs):
from .factory import resolve_delta_timestamps as _resolve_delta_timestamps
return _resolve_delta_timestamps(*args, **kwargs)
# NOTE: Low-level I/O functions (cast_stats_to_numpy, get_parquet_file_size_in_mb, etc.) # NOTE: Low-level I/O functions (cast_stats_to_numpy, get_parquet_file_size_in_mb, etc.)
# and legacy migration constants are intentionally NOT re-exported here. # and legacy migration constants are intentionally NOT re-exported here.
# Import directly: ``from lerobot.datasets.io_utils import ...`` # Import directly: ``from lerobot.datasets.io_utils import ...``