dataset = pyimport "torch/utils/data/dataset"

.VisionDataset: ClassType
.VisionDataset <: dataset.Dataset
.VisionDataset.
    __call__: (
        root: Str,
        train := Bool,
        download := Bool,
        transform := GenericCallable,
        target_transform := GenericCallable,
    ) -> .VisionDataset