node: fix encodinggetSpecialTokensMask type

This commit is contained in:
Pierric Cistac
2020-02-04 16:59:46 -05:00
parent 6165910ca6
commit 9770be5661

View File

@@ -25,7 +25,7 @@ export interface Encoding {
/**
* Returns the special tokens mask
*/
getSpecialTokensMask(): number;
getSpecialTokensMask(): number[];
/**
* Returns the tokenized string