from . import data_module
from . import pretraining_dataset


DATA_MODULES = {
    "pretrain": data_module.PretrainingDataModule,
    "pretrain_xh": data_module.PretrainingXHDataModule,
    "openi": data_module.OpenIDataModule,
    "chestxray14": data_module.ChestXray14DataModule,
    "chestxdet10": data_module.ChestXDet10DataModule,
}
