Nanobit commited on
Commit
77bdb7d
1 Parent(s): f150c02

Fix typing list

Browse files
Files changed (1) hide show
  1. src/axolotl/prompt_tokenizers.py +1 -1
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: list[int],
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
  """