diff --git a/crates/store/re_data_loader/src/lib.rs b/crates/store/re_data_loader/src/lib.rs index a80a23f32469..091be64b0b11 100644 --- a/crates/store/re_data_loader/src/lib.rs +++ b/crates/store/re_data_loader/src/lib.rs @@ -15,11 +15,11 @@ mod loader_directory; mod loader_rrd; #[cfg(not(target_arch = "wasm32"))] -mod lerobot; +pub mod lerobot; // This loader currently only works when loading the entire dataset directory, and we cannot do that on web yet. #[cfg(not(target_arch = "wasm32"))] -mod loader_lerobot; +pub mod loader_lerobot; #[cfg(not(target_arch = "wasm32"))] mod loader_external; diff --git a/crates/store/re_data_loader/src/loader_lerobot.rs b/crates/store/re_data_loader/src/loader_lerobot.rs index ddda87c5f904..b309331fc910 100644 --- a/crates/store/re_data_loader/src/loader_lerobot.rs +++ b/crates/store/re_data_loader/src/loader_lerobot.rs @@ -160,7 +160,12 @@ fn prepare_episode_chunks( store_ids } -fn load_episode( +/// Loads a single episode from a `LeRobot` dataset and converts it into a collection of Rerun chunks. +/// +/// This function processes an episode from the dataset by extracting the relevant data columns and +/// converting them into appropriate Rerun data structures. It handles different types of data +/// (videos, images, scalar values, etc.) based on their data type specifications in the dataset metadata. +pub fn load_episode( dataset: &LeRobotDataset, episode: EpisodeIndex, ) -> Result<Vec<Chunk>, DataLoaderError> {