From 120ed6453f5b917ec7007723d5ab28e33f9310e8 Mon Sep 17 00:00:00 2001 From: KerfuffleV2 Date: Sun, 27 Aug 2023 19:36:07 -0600 Subject: [PATCH] convert: Fix type hint for special_token_types in SpecialVocab --- convert.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/convert.py b/convert.py index 539ee2c41..05904209b 100755 --- a/convert.py +++ b/convert.py @@ -423,7 +423,7 @@ class SpecialVocab: special_token_types: Tuple[str, ...] = tuple(('bos', 'eos', 'unk', 'sep', 'pad')) special_token_ids: Dict[str, int] = {} - def __init__(self, path: Path, special_token_types: Optional[Tuple[str]] = None): + def __init__(self, path: Path, special_token_types: Optional[Tuple[str, ...]] = None): self.special_token_ids = {} if special_token_types is not None: self.special_token_types = special_token_types