Python - Update some missing typings

This commit is contained in:
Anthony MOI
2020-09-21 17:07:47 -04:00
committed by Anthony MOI
parent 7492a1d698
commit b9a051f464
2 changed files with 16 additions and 1 deletions

View File

@@ -1,4 +1,4 @@
from .. import Encoding, Offsets from .. import Encoding, Offsets, Token
from typing import List, Optional, Union, Tuple from typing import List, Optional, Union, Tuple
class Model: class Model:
@@ -8,6 +8,15 @@ class Model:
a Model will return a instance of this class when instantiated. a Model will return a instance of this class when instantiated.
""" """
def tokenize(self, sequence: str) -> List[Token]:
""" Tokenize the given sequence """
pass
def token_to_id(self, token: str) -> Optional[int]:
""" Returns the id associated with the given token """
pass
def id_to_token(self, id: int) -> Optional[str]:
""" Returns the token associated with the given id """
pass
def save(self, folder: str, name: Optional[str] = None) -> List[str]: def save(self, folder: str, name: Optional[str] = None) -> List[str]:
""" Save the current model """ Save the current model

View File

@@ -1,3 +1,4 @@
from .. import Encoding
from typing import Tuple, Union, List from typing import Tuple, Union, List
class PostProcessor: class PostProcessor:
@@ -14,6 +15,11 @@ class PostProcessor:
:return: :return:
""" """
pass pass
def process(
self, encoding: Encoding, pair: Optional[Encoding] = None, add_special_tokens: bool = True
) -> Encoding:
""" Post-process the given encodings, generating the final one """
pass
class BertProcessing(PostProcessor): class BertProcessing(PostProcessor):
""" BertProcessing """ BertProcessing