MinariEpisodeDataset

MinariEpisodeDataset#

class stable_pretraining.data.MinariEpisodeDataset(dataset)[source]#

Bases: Dataset

Flatten Minari episodes into individual training steps.