def get_labelled_idxs(dataset):
    """
    Get all indices with labels
    """
    split_dict = dataset.get_idx_split()
    train_idx = split_dict['train']
    valid_idx = split_dict['valid']
    test_idx = split_dict['test']
    idxs_labelled = set(train_idx)
    idxs_labelled = idxs_labelled.union(set(valid_idx))
    idxs_labelled = idxs_labelled.union(set(test_idx))
    return idxs_labelled