<filename>vision/__init__.py
from .datasets import StanfordCars
from .datasets import CelebADataset

Datasets = {
    "celeba": CelebADataset,
    'stanfordcars': StanfordCars,
}

<filename>vision/__init__.py
from .datasets import StanfordCars
from .datasets import CelebADataset

Datasets = {
    "celeba": CelebADataset,
    'stanfordcars': StanfordCars,
}
