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 * Returns the special tokens mask
*/ */
getSpecialTokensMask(): number; getSpecialTokensMask(): number[];
/** /**
* Returns the tokenized string * Returns the tokenized string