from .data import LitMNISTDataModule