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 |
"""
|