cyto_dl.utils.dummy_dataset module

class cyto_dl.utils.dummy_dataset.DummyDatamodule(num_samples, batch_size, shapes, dummy_metadata=None, **kwargs)

Bases: LightningDataModule

get_dataloader()

predict_dataloader()

test_dataloader()

train_dataloader()

val_dataloader()

class cyto_dl.utils.dummy_dataset.DummyDataset(num_samples: int = 10000, dummy_metadata: dict | None = None, **shapes)

Bases: Dataset

Args:

num_samples: how many samples to use in this dataset metadata: whether to generate metatensors **shapes: kwargs, where each key will become a batch key and each

value is the shape of the corresponding batch elements

generate_img(k)