def METHOD_NAME(
    batch_size,
    data_dir="./data-test/cifar10",
    download=True,
    transform=None,
    source_url=None,
    num_workers=0,
):
    cifar10_train = vision.datasets.CIFAR10(
        root=data_dir,
        train=True,
        download=download,
        transform=transform,
        source_url=source_url,
    )
    cifar10_test = vision.datasets.CIFAR10(
        root=data_dir,
        train=False,
        download=download,
        transform=transform,
        source_url=source_url,
    )
    train_iter = flow.utils.data.DataLoader(
        cifar10_train, batch_size=batch_size, shuffle=True, num_workers=num_workers
    )
    test_iter = flow.utils.data.DataLoader(
        cifar10_test, batch_size=batch_size, shuffle=False, num_workers=num_workers
    )
    return train_iter, test_iter