def METHOD_NAME(dataset):
    tab = wandb.Table(
        data=[
            [str(np.random.choice(range(dataset_size)).tolist()), i, random_image()]
            for i in range(pred_size)
        ],
        columns=["fk_id", "tab_id", "pred_img"],
    )
    tab.set_fk("fk_id", dataset, "id")
    return tab