15 lines
426 B
Python
15 lines
426 B
Python
from postal.text.encoding import safe_encode, safe_decode
|
|
from postal.text import _tokenize
|
|
from postal.text.token_types import token_types
|
|
|
|
|
|
def tokenize_raw(s):
|
|
return _tokenize.tokenize(safe_decode(s))
|
|
|
|
|
|
def tokenize(s):
|
|
u = safe_decode(s)
|
|
s = safe_encode(s)
|
|
return [(safe_decode(s[start:start + length]), token_types.from_id(token_type))
|
|
for start, length, token_type in _tokenize.tokenize(u)]
|