def data_prep_function(data, candidates, pixel_spacing, p_transform,
                       p_transform_augment, **kwargs):
    x = data_transforms.build_dsb_can_heatmap(data=data,
                                                 candidates=candidates,
                                                 p_transform=p_transform,
                                                 p_transform_augment=p_transform_augment,
                                                 pixel_spacing=pixel_spacing)

    return x
def data_prep_function(data, candidates, pixel_spacing, p_transform,
                       p_transform_augment, **kwargs):
    x = data_transforms.build_dsb_can_heatmap(
        data=data,
        candidates=candidates,
        p_transform=p_transform,
        p_transform_augment=p_transform_augment,
        pixel_spacing=pixel_spacing)

    return x