Working with Studio Datasets

Studio datasets use the LeRobot format and are consumed through Lightning datamodules. The datamodules are first-class Python API objects; YAML/CLI configs are a serialization of the same construction path. These modules ship with physicalai-train:

Key modules:

• data/lerobot/datamodule.py — LeRobotDataModule (the class configs reference as physicalai.data.lerobot.LeRobotDataModule).
• data/lerobot/dataset.py — LeRobot dataset wrapper.
• data/lerobot/converters.py — DataFormat and public FormatConverter.to_lerobot_dict(...) / to_observation(...).
• data/observation.py — Observation, Feature, FeatureType, NormalizationParameters (match your policy's features).
• data/datamodules.py — base DataModule (Lightning LightningDataModule, auto num-workers heuristic).
• data/dataset.py — base Dataset; data/gym.py — GymDataset for gym-generated data.

Python API usage

Use this path for notebooks, tests, direct batch inspection, or debugging dataloading without involving the training CLI.

from physicalai.data import LeRobotDataModule