def METHOD_NAME(dataset, sampler_option="random", n_bins=5):
    df = dataset.df
    weights = [1] * len(df) * dataset.elem_per_image
    if sampler_option == "random":
        return sampler.RandomSampler(weights)
    else:
        raise NotImplementedError(
            f"The option {sampler_option} for sampler on reconstruction task is not implemented"
        )