Word indices are None for special tokens

This commit is contained in:
Anthony MOI
2020-04-09 09:52:02 -04:00
parent e7949fcb84
commit 3ad1360210
10 changed files with 66 additions and 74 deletions

View File

@ -62,7 +62,7 @@ impl Encoding {
}
#[getter]
fn get_words(&self) -> Vec<u32> {
fn get_words(&self) -> Vec<Option<u32>> {
self.encoding.get_words().to_vec()
}