docs and examples imports update

examples/dataset/load_lerobot_dataset.py

@@ -31,8 +31,7 @@ from pprint import pprint
 import torch
 from huggingface_hub import HfApi
 
-from lerobot.datasets.dataset_metadata import LeRobotDatasetMetadata
-from lerobot.datasets.lerobot_dataset import LeRobotDataset
+from lerobot.datasets import LeRobotDataset, LeRobotDatasetMetadata
 
 
 def main():

examples/dataset/slurm_compute_rabc.py

@@ -231,7 +231,7 @@ class AggregateProgress(PipelineStep):
         import pyarrow as pa
         import pyarrow.parquet as pq
 
-        from lerobot.datasets.lerobot_dataset import LeRobotDataset
+        from lerobot.datasets import LeRobotDataset
         from lerobot.utils.utils import init_logging
 
         init_logging()

examples/dataset/use_dataset_image_transforms.py

@@ -26,7 +26,7 @@ import torch
 from torchvision.transforms import v2
 from torchvision.transforms.functional import to_pil_image
 
-from lerobot.datasets.lerobot_dataset import LeRobotDataset
+from lerobot.datasets import LeRobotDataset
 from lerobot.transforms import ImageTransformConfig, ImageTransforms, ImageTransformsConfig
 
 

examples/dataset/use_dataset_tools.py

@@ -29,15 +29,14 @@ Usage:
 
 import numpy as np
 
-from lerobot.datasets.dataset_tools import (
-    add_features,
+from lerobot.datasets import (
+    LeRobotDataset,
     delete_episodes,
     merge_datasets,
-    modify_features,
     remove_feature,
     split_dataset,
 )
-from lerobot.datasets.lerobot_dataset import LeRobotDataset
+from lerobot.datasets.dataset_tools import add_features, modify_features
 
 
 def main():