feat(dependencies): minimal default tag install (#3362)

examples/dataset/load_lerobot_dataset.py

@@ -31,17 +31,11 @@ from pprint import pprint
 import torch
 from huggingface_hub import HfApi
 
-import lerobot
-from lerobot.datasets.dataset_metadata import LeRobotDatasetMetadata
-from lerobot.datasets.lerobot_dataset import LeRobotDataset
+from lerobot.datasets import LeRobotDataset, LeRobotDatasetMetadata
 
 
 def main():
-    # We ported a number of existing datasets ourselves, use this to see the list:
-    print("List of available datasets:")
-    pprint(lerobot.available_datasets)
-
-    # You can also browse through the datasets created/ported by the community on the hub using the hub api:
+    # Browse datasets created/ported by the community on the hub using the hub api:
     hub_api = HfApi()
     repo_ids = [info.id for info in hub_api.list_datasets(task_categories="robotics", tags=["LeRobot"])]
     pprint(repo_ids)

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,8 +26,8 @@ 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.transforms import ImageTransformConfig, ImageTransforms, ImageTransformsConfig
+from lerobot.datasets import LeRobotDataset
+from lerobot.transforms import ImageTransformConfig, ImageTransforms, ImageTransformsConfig
 
 
 def save_image(tensor, filename):

examples/dataset/use_dataset_tools.py

@@ -29,7 +29,8 @@ Usage:
 
 import numpy as np
 
-from lerobot.datasets.dataset_tools import (
+from lerobot.datasets import (
+    LeRobotDataset,
     add_features,
     delete_episodes,
     merge_datasets,
@@ -37,7 +38,6 @@ from lerobot.datasets.dataset_tools import (
     remove_feature,
     split_dataset,
 )
-from lerobot.datasets.lerobot_dataset import LeRobotDataset
 
 
 def main():