torchcell.datamodules.DataModuleIndex

class DataModuleIndex(*, train: list[int], val: list[int], test: list[int])

Bases: ModelStrict

Train/val/test index lists that are sorted and mutually non-overlapping.

classmethod check_sorted_and_unique_indices(values: dict[str, Any]) → dict[str, Any]

Validate each split is sorted and the splits do not overlap.

model_config: ClassVar[ConfigDict] = {'extra': 'forbid', 'frozen': True}

Configuration for the model, should be a dictionary conforming to [ConfigDict][pydantic.config.ConfigDict].