File size: 305 Bytes
626eca0
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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) -> List[str]:
    return [NME_SYMBOL] + [f"[R-{i}]" for i in range(num_entities)]