Module lerobot

A module for loading and working with LeRobot datasets.

This module provides functionality to identify and parse LeRobot datasets, which consist of metadata and episode data stored in a structured format.

Important

Currently this only supports v2 LeRobot datasets!

See LeRobotDataset for more information on the dataset format.

Structs

• EpisodeIndex
  Newtype wrapper for episode indices.
• Feature
  Feature definition for a LeRobot dataset.
• LeRobotDataset
  A LeRobot dataset consists of structured metadata and recorded episode data stored in Parquet files.
• LeRobotDatasetEpisode
  An episode in a LeRobot dataset.
• LeRobotDatasetInfo
  LeRobot dataset metadata.
• LeRobotDatasetMetadata
  Metadata for a LeRobot dataset.
• LeRobotDatasetTask
  A task in a LeRobot dataset.
• Names
  Name metadata for a feature in the LeRobot dataset.
• NamesVisitor 🔒
  Visitor implementation for deserializing the Names type.
• TaskIndex
  Newtype wrapper for task indices.

Enums

• DType
  Data types supported for features in a LeRobot dataset.
• LeRobotError
  Errors that might happen when loading data through a crate::loader_lerobot::LeRobotDatasetLoader.

Functions

• has_sub_directories 🔒
• is_lerobot_dataset
  Check whether the provided path contains a LeRobot dataset.
• is_v1_lerobot_dataset
  Check whether the provided path contains a v1 LeRobot dataset.
• is_v2_lerobot_dataset
  Check whether the provided path contains a v2 LeRobot dataset.
• load_jsonl_file 🔒