from typing import List

NME_SYMBOL = "--NME--"


def get_special_symbols(num_entities: int) -> List[str]:
    return [NME_SYMBOL] + [f"[E-{i}]" for i in range(num_entities)]


def get_special_symbols_re(num_entities: int, use_nme: bool = False) -> List[str]:
    if use_nme:
        return [NME_SYMBOL] + [f"[R-{i}]" for i in range(num_entities)]
    else:
        return [f"[R-{i}]" for i in range(num_entities)]