2025ML-project-neural_compression/CNN-model/dataset_loaders/__init__.py

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

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