def get_bert_dataset_name(is_cased):
    """Returns relevant BERT dataset name, depending on whether we are using a cased model.

    Parameters
    ----------
    is_cased: bool
        Whether we are using a cased model.

    Returns
    -------
    str: Named of the BERT dataset.

    """
    if is_cased:
        return 'book_corpus_wiki_en_cased'
    else:
        return 'book_corpus_wiki_en_uncased'