diff --git a/src/torchaudio/pipelines/_wav2vec2/aligner.py b/src/torchaudio/pipelines/_wav2vec2/aligner.py index 3655d5bae8..bf75f5f93e 100644 --- a/src/torchaudio/pipelines/_wav2vec2/aligner.py +++ b/src/torchaudio/pipelines/_wav2vec2/aligner.py @@ -9,7 +9,7 @@ class ITokenizer(ABC): @abstractmethod - def __call__(self, transcript: List[str]) -> List[List[str]]: + def __call__(self, transcript: List[str]) -> List[List[int]]: """Tokenize the given transcript (list of word) .. note::