From d3ad24d9ddbf1df0687b4893ca3ef54439f4dec1 Mon Sep 17 00:00:00 2001 From: Pepijn Date: Wed, 15 Jul 2026 15:12:17 +0200 Subject: [PATCH] revert(datasets): use main package exports --- src/lerobot/datasets/__init__.py | 20 +------------------- 1 file changed, 1 insertion(+), 19 deletions(-) diff --git a/src/lerobot/datasets/__init__.py b/src/lerobot/datasets/__init__.py index 35c822b2f..7715a115e 100644 --- a/src/lerobot/datasets/__init__.py +++ b/src/lerobot/datasets/__init__.py @@ -35,6 +35,7 @@ from .dataset_tools import ( remove_feature, split_dataset, ) +from .factory import make_dataset, make_train_eval_datasets, resolve_delta_timestamps from .image_writer import safe_stop_image_writer from .io_utils import load_episodes, write_stats from .language import ( @@ -54,25 +55,6 @@ from .streaming_dataset import StreamingLeRobotDataset from .utils import DEFAULT_EPISODES_PATH, create_lerobot_dataset_card 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.) # and legacy migration constants are intentionally NOT re-exported here. # Import directly: ``from lerobot.datasets.io_utils import ...``