from .Dataset import Dataset
from .EnWik9 import EnWik9DataSet
from .LoremIpsumDataset import LoremIpsumDataset

dataset_called: dict[str, type[Dataset]] = {
    'enwik9': EnWik9DataSet,
    'lorem_ipsum': LoremIpsumDataset
}
