torchcell.datasets.dataset_registry.register_dataset

register_dataset(cls: T) → T[source]

Register a dataset class by its name and return it unchanged (decorator).