Fix typing list
Browse files
src/axolotl/prompt_tokenizers.py
CHANGED
@@ -440,7 +440,7 @@ def parse_tokenized_to_result(
|
|
440 |
result: Dict[str, List[int]],
|
441 |
current_len: int,
|
442 |
res: Dict[str, List[int]],
|
443 |
-
labels:
|
444 |
pad_token_id: Union[int, None] = None,
|
445 |
) -> Tuple[Dict[str, List[int]], int]:
|
446 |
"""
|
|
|
440 |
result: Dict[str, List[int]],
|
441 |
current_len: int,
|
442 |
res: Dict[str, List[int]],
|
443 |
+
labels: List[int],
|
444 |
pad_token_id: Union[int, None] = None,
|
445 |
) -> Tuple[Dict[str, List[int]], int]:
|
446 |
"""
|