Fix type bug

This commit is contained in:
Samuel Sloniker 2023-06-15 12:15:39 -07:00
parent a1a200dd58
commit 79d8c73183

View File

@ -1,7 +1,7 @@
# SPDX-License-Identifier: GPL-3.0-or-later
import unicodedata
from typing import List, Iterator, Iterable
from typing import Iterator, Iterable
import emoji
@ -46,7 +46,7 @@ def tokenize(
yield last_token
def ngrams(tokens: Iterable[str], max_ngram_length: int) -> List[str]:
def ngrams(tokens: Iterable[str], max_ngram_length: int) -> Iterable[str]:
if max_ngram_length == 1:
return tokens