Given an index, convert arbitrary data into a State or a list of States.
State
BaseDataset.arbitrary_to_ds()
dgs.models.dataset.dataset.BaseDataset
dgs.models.dataset.dataset.BaseDataset.configure_torch_module