File size: 415 Bytes
2f044c1 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 |
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)]
|