def __init__()

in torchrecipes/vision/data/modules/mnist_data_module.py [0:0]

• 14 lines of code
• 1 McCabe index (conditional complexity)

    def __init__(
        self,
        data_dir: Optional[str] = None,
        val_split: Union[int, float] = 0.2,
        num_workers: int = 16,
        normalize: bool = False,
        batch_size: int = 32,
        seed: int = 42,
        shuffle: bool = False,
        pin_memory: bool = False,
        drop_last: bool = False,
        train_transforms: Optional[Callable] = None,  # pyre-ignore[24]
        val_transforms: Optional[Callable] = None,  # pyre-ignore[24]
        test_transforms: Optional[Callable] = None,  # pyre-ignore[24]