



"""simple docstring"""
import unittest

from transformers import SPIECE_UNDERLINE
from transformers.models.speechta import SpeechTaTokenizer
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_tokenizers, slow
from transformers.tokenization_utils import AddedToken

from ...test_tokenization_common import TokenizerTesterMixin


SCREAMING_SNAKE_CASE_				= get_tests_dir("""fixtures/test_sentencepiece_bpe_char.model""")




@require_sentencepiece
@require_tokenizers
class 			snake_case_						(						a_     ,unittest.TestCase				):
							__lowerCAmelCase    					=					SpeechTaTokenizer
							__lowerCAmelCase    					=					False
							__lowerCAmelCase    					=					True






							def snake_case_  (   self      ):
									super().setUp()

									# We have a SentencePiece fixture for testing
									a_     :      Any 						=     SpeechTaTokenizer(a_      )

									a_     :      Optional[int] 						=     AddedToken("<mask>"			,     lstrip=a_			,     rstrip=a_      )
									a_     :      Any 						=     mask_token
									tokenizer.add_special_tokens({"mask_token": mask_token}      )
									tokenizer.add_tokens(["<ctc_blank>"]      )

									tokenizer.save_pretrained(self.tmpdirname      )






							def snake_case_  (   self			,     a_      ):
									a_     :      Tuple 						=     "this is a test"
									a_     :      Any 						=     "this is a test"
									return input_text, output_text






							def snake_case_  (   self			,     a_			,     a_=False			,     a_=2_0			,     a_=5      ):
									a_						,				a_     :      Optional[Any] 						=     self.get_input_output_texts(a_      )
									a_     :      Optional[Any] 						=     tokenizer.encode(a_			,     add_special_tokens=a_      )
									a_     :      Dict 						=     tokenizer.decode(a_			,     clean_up_tokenization_spaces=a_      )
									return text, ids






							def snake_case_  (   self      ):
									a_     :      List[Any] 						=     "<pad>"
									a_     :      Optional[int] 						=     1

									self.assertEqual(self.get_tokenizer()._convert_token_to_id(a_      )			,     a_      )
									self.assertEqual(self.get_tokenizer()._convert_id_to_token(a_      )			,     a_      )






							def snake_case_  (   self      ):
									a_     :      List[Any] 						=     list(self.get_tokenizer().get_vocab().keys()      )

									self.assertEqual(vocab_keys[0]			,     "<s>"      )
									self.assertEqual(vocab_keys[1]			,     "<pad>"      )
									self.assertEqual(vocab_keys[-4]			,     "œ"      )
									self.assertEqual(vocab_keys[-2]			,     "<mask>"      )
									self.assertEqual(vocab_keys[-1]			,     "<ctc_blank>"      )
									self.assertEqual(len(a_      )			,     8_1      )






							def snake_case_  (   self      ):
									self.assertEqual(self.get_tokenizer().vocab_size			,     7_9      )






							def snake_case_  (   self      ):
									a_     :      Any 						=     self.get_tokenizers(do_lower_case=a_      )
									for tokenizer in tokenizers:
											with self.subTest(F"""{tokenizer.__class__.__name__}"""      ):
													a_     :      Dict 						=     tokenizer.vocab_size
													a_     :      List[str] 						=     len(a_      )

													self.assertNotEqual(a_			,     0      )

													# We usually have added tokens from the start in tests because our vocab fixtures are
													# smaller than the original vocabs - let's not assert this
													# self.assertEqual(vocab_size, all_size)

													a_     :      Optional[int] 						=     ["aaaaa bbbbbb", "cccccccccdddddddd"]
													a_     :      int 						=     tokenizer.add_tokens(a_      )
													a_     :      List[Any] 						=     tokenizer.vocab_size
													a_     :      Tuple 						=     len(a_      )

													self.assertNotEqual(a_			,     0      )
													self.assertEqual(a_			,     a_      )
													self.assertEqual(a_			,     len(a_      )      )
													self.assertEqual(a_			,     all_size + len(a_      )      )

													a_     :      str 						=     tokenizer.encode("aaaaa bbbbbb low cccccccccdddddddd l"			,     add_special_tokens=a_      )

													self.assertGreaterEqual(len(a_      )			,     4      )
													self.assertGreater(tokens[0]			,     tokenizer.vocab_size - 1      )
													self.assertGreater(tokens[-3]			,     tokenizer.vocab_size - 1      )

													a_     :      Tuple 						=     {"eos_token": ">>>>|||<||<<|<<", "pad_token": "<<<<<|||>|>>>>|>"}
													a_     :      Dict 						=     tokenizer.add_special_tokens(a_      )
													a_     :      Optional[Any] 						=     tokenizer.vocab_size
													a_     :      Any 						=     len(a_      )

													self.assertNotEqual(a_			,     0      )
													self.assertEqual(a_			,     a_      )
													self.assertEqual(a_			,     len(a_      )      )
													self.assertEqual(a_			,     all_size_a + len(a_      )      )

													a_     :      Any 						=     tokenizer.encode(
													    ">>>>|||<||<<|<< aaaaabbbbbb low cccccccccdddddddd <<<<<|||>|>>>>|> l"			,     add_special_tokens=a_      )

													self.assertGreaterEqual(len(a_      )			,     6      )
													self.assertGreater(tokens[0]			,     tokenizer.vocab_size - 1      )
													self.assertGreater(tokens[0]			,     tokens[1]      )
													self.assertGreater(tokens[-3]			,     tokenizer.vocab_size - 1      )
													self.assertGreater(tokens[-3]			,     tokens[-4]      )
													self.assertEqual(tokens[0]			,     tokenizer.eos_token_id      )
													self.assertEqual(tokens[-3]			,     tokenizer.pad_token_id      )






							def snake_case_  (   self      ):
									pass






							def snake_case_  (   self      ):
									pass






							def snake_case_  (   self      ):
									a_     :      Union[str, Any] 						=     self.get_tokenizer()

									a_     :      Any 						=     tokenizer.tokenize("This is a test"      )
									# fmt: off
									self.assertListEqual(a_			,     [SPIECE_UNDERLINE, "T", "h", "i", "s", SPIECE_UNDERLINE, "i", "s", SPIECE_UNDERLINE, "a", SPIECE_UNDERLINE, "t", "e", "s", "t"]      )
									# fmt: on

									self.assertListEqual(
									    tokenizer.convert_tokens_to_ids(a_      )			,     [4, 3_2, 1_1, 1_0, 1_2, 4, 1_0, 1_2, 4, 7, 4, 6, 5, 1_2, 6]			,     )

									a_     :      Optional[int] 						=     tokenizer.tokenize("I was born in 92000, and this is falsé."      )
									self.assertListEqual(
									    a_			,     [SPIECE_UNDERLINE, "I", SPIECE_UNDERLINE, "w", "a", "s", SPIECE_UNDERLINE, "b", "o", "r", "n", SPIECE_UNDERLINE, "i", "n", SPIECE_UNDERLINE, "92000", ",", SPIECE_UNDERLINE, "a", "n", "d", SPIECE_UNDERLINE, "t", "h", "i", "s", SPIECE_UNDERLINE, "i", "s", SPIECE_UNDERLINE, "f", "a", "l", "s", "é", "."]      )

									a_     :      Tuple 						=     tokenizer.convert_tokens_to_ids(a_      )
									# fmt: off
									self.assertListEqual(a_			,     [4, 3_0, 4, 2_0, 7, 1_2, 4, 2_5, 8, 1_3, 9, 4, 1_0, 9, 4, 3, 2_3, 4, 7, 9, 1_4, 4, 6, 1_1, 1_0, 1_2, 4, 1_0, 1_2, 4, 1_9, 7, 1_5, 1_2, 7_3, 2_6]      )
									# fmt: on

									a_     :      Tuple 						=     tokenizer.convert_ids_to_tokens(a_      )
									self.assertListEqual(
									    a_			,     [SPIECE_UNDERLINE, "I", SPIECE_UNDERLINE, "w", "a", "s", SPIECE_UNDERLINE, "b", "o", "r", "n", SPIECE_UNDERLINE, "i", "n", SPIECE_UNDERLINE, "<unk>", ",", SPIECE_UNDERLINE, "a", "n", "d", SPIECE_UNDERLINE, "t", "h", "i", "s", SPIECE_UNDERLINE, "i", "s", SPIECE_UNDERLINE, "f", "a", "l", "s", "é", "."]      )






							@slow
							def snake_case_  (   self      ):
									# Use custom sequence because this tokenizer does not handle numbers.
									a_     :      List[Any] 						=     [
									    "Transformers (formerly known as pytorch-transformers and pytorch-pretrained-bert) provides "
									    "general-purpose architectures (BERT, GPT, RoBERTa, XLM, DistilBert, XLNet...) for Natural "
									    "Language Understanding (NLU) and Natural Language Generation (NLG) with over thirty-two pretrained "
									    "models in one hundred plus languages and deep interoperability between Jax, PyTorch and TensorFlow.",
									    "BERT is designed to pre-train deep bidirectional representations from unlabeled text by jointly "
									    "conditioning on both left and right context in all layers.",
									    "The quick brown fox jumps over the lazy dog.",
									]

									# fmt: off
									a_     :      Tuple 						=     {
									    "input_ids": [
									        [4, 3_2, 1_3, 7, 9, 1_2, 1_9, 8, 1_3, 1_8, 5, 1_3, 1_2, 4, 6_4, 1_9, 8, 1_3, 1_8, 5, 1_3, 1_5, 2_2, 4, 2_8, 9, 8, 2_0, 9, 4, 7, 1_2, 4, 2_4, 2_2, 6, 8, 1_3, 1_7, 1_1, 3_9, 6, 1_3, 7, 9, 1_2, 1_9, 8, 1_3, 1_8, 5, 1_3, 1_2, 4, 7, 9, 1_4, 4, 2_4, 2_2, 6, 8, 1_3, 1_7, 1_1, 3_9, 2_4, 1_3, 5, 6, 1_3, 7, 1_0, 9, 5, 1_4, 3_9, 2_5, 5, 1_3, 6, 6_3, 4, 2_4, 1_3, 8, 2_7, 1_0, 1_4, 5, 1_2, 4, 2_1, 5, 9, 5, 1_3, 7, 1_5, 3_9, 2_4, 1_6, 1_3, 2_4, 8, 1_2, 5, 4, 7, 1_3, 1_7, 1_1, 1_0, 6, 5, 1_7, 6, 1_6, 1_3, 5, 1_2, 4, 6_4, 4_0, 4_7, 5_4, 3_2, 2_3, 4, 5_3, 4_9, 3_2, 2_3, 4, 5_4, 8, 4_0, 4_7, 5_4, 3_2, 7, 2_3, 4, 6_9, 5_2, 4_3, 2_3, 4, 5_1, 1_0, 1_2, 6, 1_0, 1_5, 4_0, 5, 1_3, 6, 2_3, 4, 6_9, 5_2, 4_8, 5, 6, 2_6, 2_6, 2_6, 6_3, 4, 1_9, 8, 1_3, 4, 4_8, 7, 6, 1_6, 1_3, 7, 1_5, 4, 5_2, 7, 9, 2_1, 1_6, 7, 2_1, 5, 4, 6_1, 9, 1_4, 5, 1_3, 1_2, 6, 7, 9, 1_4, 1_0, 9, 2_1, 4, 6_4, 4_8, 5_2, 6_1, 6_3, 4, 7, 9, 1_4, 4, 4_8, 7, 6, 1_6, 1_3, 7, 1_5, 4, 5_2, 7, 9, 2_1, 1_6, 7, 2_1, 5, 4, 5_3, 5, 9, 5, 1_3, 7, 6, 1_0, 8, 9, 4, 6_4, 4_8, 5_2, 5_3, 6_3, 4, 2_0, 1_0, 6, 1_1, 4, 8, 2_7, 5, 1_3, 4, 6, 1_1, 1_0, 1_3, 6, 2_2, 3_9, 6, 2_0, 8, 4, 2_4, 1_3, 5, 6, 1_3, 7, 1_0, 9, 5, 1_4, 4, 1_8, 8, 1_4, 5, 1_5, 1_2, 4, 1_0, 9, 4, 8, 9, 5, 4, 1_1, 1_6, 9, 1_4, 1_3, 5, 1_4, 4, 2_4, 1_5, 1_6, 1_2, 4, 1_5, 7, 9, 2_1, 1_6, 7, 2_1, 5, 1_2, 4, 7, 9, 1_4, 4, 1_4, 5, 5, 2_4, 4, 1_0, 9, 6, 5, 1_3, 8, 2_4, 5, 1_3, 7, 2_5, 1_0, 1_5, 1_0, 6, 2_2, 4, 2_5, 5, 6, 2_0, 5, 5, 9, 4, 5_8, 7, 3_7, 2_3, 4, 4_9, 2_2, 3_2, 8, 1_3, 1_7, 1_1, 4, 7, 9, 1_4, 4, 3_2, 5, 9, 1_2, 8, 1_3, 5_5, 1_5, 8, 2_0, 2_6, 2],
									        [4, 4_0, 4_7, 5_4, 3_2, 4, 1_0, 1_2, 4, 1_4, 5, 1_2, 1_0, 2_1, 9, 5, 1_4, 4, 6, 8, 4, 2_4, 1_3, 5, 3_9, 6, 1_3, 7, 1_0, 9, 4, 1_4, 5, 5, 2_4, 4, 2_5, 1_0, 1_4, 1_0, 1_3, 5, 1_7, 6, 1_0, 8, 9, 7, 1_5, 4, 1_3, 5, 2_4, 1_3, 5, 1_2, 5, 9, 6, 7, 6, 1_0, 8, 9, 1_2, 4, 1_9, 1_3, 8, 1_8, 4, 1_6, 9, 1_5, 7, 2_5, 5, 1_5, 5, 1_4, 4, 6, 5, 3_7, 6, 4, 2_5, 2_2, 4, 4_6, 8, 1_0, 9, 6, 1_5, 2_2, 4, 1_7, 8, 9, 1_4, 1_0, 6, 1_0, 8, 9, 1_0, 9, 2_1, 4, 8, 9, 4, 2_5, 8, 6, 1_1, 4, 1_5, 5, 1_9, 6, 4, 7, 9, 1_4, 4, 1_3, 1_0, 2_1, 1_1, 6, 4, 1_7, 8, 9, 6, 5, 3_7, 6, 4, 1_0, 9, 4, 7, 1_5, 1_5, 4, 1_5, 7, 2_2, 5, 1_3, 1_2, 2_6, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
									        [4, 3_2, 1_1, 5, 4, 4_5, 1_6, 1_0, 1_7, 2_8, 4, 2_5, 1_3, 8, 2_0, 9, 4, 1_9, 8, 3_7, 4, 4_6, 1_6, 1_8, 2_4, 1_2, 4, 8, 2_7, 5, 1_3, 4, 6, 1_1, 5, 4, 1_5, 7, 5_7, 2_2, 4, 1_4, 8, 2_1, 2_6, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
									    ],
									    "attention_mask": [
									        [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1],
									        [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
									        [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
									    ]
									}
									# fmt: on

									self.tokenizer_integration_test_util(
									    expected_encoding=a_			,     model_name="microsoft/speecht5_asr"			,     revision="c5ef64c71905caeccde0e4462ef3f9077224c524"			,     sequences=a_			,     )




"""simple docstring"""
from collections import OrderedDict
from typing import Any, Mapping, Optional

from ... import PreTrainedTokenizer
from ...configuration_utils import PretrainedConfig
from ...file_utils import TensorType, is_torch_available
from ...onnx import OnnxConfig, OnnxConfigWithPast, OnnxSeqaSeqConfigWithPast
from ...onnx.utils import compute_effective_axis_dimension
from ...utils import logging


SCREAMING_SNAKE_CASE_				= logging.get_logger(__name__)

SCREAMING_SNAKE_CASE_				= {
    """facebook/blenderbot_small-90M""": """https://huggingface.co/facebook/blenderbot_small-90M/resolve/main/config.json""",
    # See all BlenderbotSmall models at https://huggingface.co/models?filter=blenderbot_small
}




class 			snake_case_						(						a_				):
							__lowerCAmelCase    					=					"blenderbot-small"
							__lowerCAmelCase    					=					["past_key_values"]
							__lowerCAmelCase    					=					{"num_attention_heads": "encoder_attention_heads", "hidden_size": "d_model"}






							def __init__(   self			,     a_=5_0_2_6_5			,     a_=5_1_2			,     a_=8			,     a_=2_0_4_8			,     a_=1_6			,     a_=8			,     a_=2_0_4_8			,     a_=1_6			,     a_=0.0			,     a_=0.0			,     a_=True			,     a_=True			,     a_="gelu"			,     a_=5_1_2			,     a_=0.1			,     a_=0.0			,     a_=0.0			,     a_=0.02			,     a_=1			,     a_=False			,     a_=0			,     a_=1			,     a_=2			,     a_=2			,     **a_			,     ):
									a_     :      int 						=     vocab_size
									a_     :      Any 						=     max_position_embeddings
									a_     :      Optional[int] 						=     d_model
									a_     :      Tuple 						=     encoder_ffn_dim
									a_     :      List[Any] 						=     encoder_layers
									a_     :      Optional[int] 						=     encoder_attention_heads
									a_     :      Optional[int] 						=     decoder_ffn_dim
									a_     :      List[str] 						=     decoder_layers
									a_     :      Dict 						=     decoder_attention_heads
									a_     :      List[str] 						=     dropout
									a_     :      List[Any] 						=     attention_dropout
									a_     :      List[str] 						=     activation_dropout
									a_     :      Optional[Any] 						=     activation_function
									a_     :      List[Any] 						=     init_std
									a_     :      int 						=     encoder_layerdrop
									a_     :      Optional[int] 						=     decoder_layerdrop
									a_     :      List[str] 						=     use_cache
									a_     :      Optional[int] 						=     encoder_layers
									a_     :      Optional[int] 						=     scale_embedding  # scale factor will be sqrt(d_model) if True

									super().__init__(
									    pad_token_id=a_			,     bos_token_id=a_			,     eos_token_id=a_			,     is_encoder_decoder=a_			,     decoder_start_token_id=a_			,     forced_eos_token_id=a_			,     **a_			,     )




class 			snake_case_						(						a_				):






							@property
							def snake_case_  (   self      ):
									if self.task in ["default", "seq2seq-lm"]:
											a_     :      Optional[int] 						=     OrderedDict(
											    [
											        ("input_ids", {0: "batch", 1: "encoder_sequence"}),
											        ("attention_mask", {0: "batch", 1: "encoder_sequence"}),
											    ]      )

											if self.use_past:
													a_     :      Tuple 						=     {0: "batch"}
													a_     :      int 						=     {0: "batch", 1: "past_decoder_sequence + sequence"}
											else:
													a_     :      List[str] 						=     {0: "batch", 1: "decoder_sequence"}
													a_     :      Optional[int] 						=     {0: "batch", 1: "decoder_sequence"}

											if self.use_past:
													self.fill_with_past_key_values_(a_			,     direction="inputs"      )
									elif self.task == "causal-lm":
											# TODO: figure this case out.
											a_     :      Dict 						=     OrderedDict(
											    [
											        ("input_ids", {0: "batch", 1: "encoder_sequence"}),
											        ("attention_mask", {0: "batch", 1: "encoder_sequence"}),
											    ]      )
											if self.use_past:
													a_						,				a_     :      Optional[int] 						=     self.num_layers
													for i in range(a_      ):
															a_     :      Union[str, Any] 						=     {0: "batch", 2: "past_sequence + sequence"}
															a_     :      List[Any] 						=     {0: "batch", 2: "past_sequence + sequence"}
									else:
											a_     :      List[str] 						=     OrderedDict(
											    [
											        ("input_ids", {0: "batch", 1: "encoder_sequence"}),
											        ("attention_mask", {0: "batch", 1: "encoder_sequence"}),
											        ("decoder_input_ids", {0: "batch", 1: "decoder_sequence"}),
											        ("decoder_attention_mask", {0: "batch", 1: "decoder_sequence"}),
											    ]      )

									return common_inputs






							@property
							def snake_case_  (   self      ):
									if self.task in ["default", "seq2seq-lm"]:
											a_     :      List[Any] 						=     super().outputs
									else:
											a_     :      Tuple 						=     super(a_			,     self      ).outputs
											if self.use_past:
													a_						,				a_     :      Dict 						=     self.num_layers
													for i in range(a_      ):
															a_     :      List[Any] 						=     {0: "batch", 2: "past_sequence + sequence"}
															a_     :      List[Any] 						=     {0: "batch", 2: "past_sequence + sequence"}
									return common_outputs






							def snake_case_  (   self			,     a_			,     a_ = -1			,     a_ = -1			,     a_ = False			,     a_ = None			,     ):
									a_     :      List[str] 						=     self._generate_dummy_inputs_for_sequence_classification_and_question_answering(
									    a_			,     a_			,     a_			,     a_			,     a_      )

									# Generate decoder inputs
									a_     :      Optional[int] 						=     seq_length if not self.use_past else 1
									a_     :      str 						=     self._generate_dummy_inputs_for_sequence_classification_and_question_answering(
									    a_			,     a_			,     a_			,     a_			,     a_      )
									a_     :      int 						=     {F"""decoder_{name}""": tensor for name, tensor in decoder_inputs.items()}
									a_     :      Tuple 						=     dict(**a_			,     **a_      )

									if self.use_past:
											if not is_torch_available():
													raise ValueError("Cannot generate dummy past_keys inputs without PyTorch installed."      )
											else:
													import torch
											a_						,				a_     :      Optional[int] 						=     common_inputs["input_ids"].shape
											a_     :      str 						=     common_inputs["decoder_input_ids"].shape[1]
											a_						,				a_     :      Dict 						=     self.num_attention_heads
											a_     :      List[str] 						=     (
											    batch,
											    num_encoder_attention_heads,
											    encoder_seq_length,
											    self._config.hidden_size // num_encoder_attention_heads,
											)
											a_     :      Optional[Any] 						=     decoder_seq_length + 3
											a_     :      str 						=     (
											    batch,
											    num_decoder_attention_heads,
											    decoder_past_length,
											    self._config.hidden_size // num_decoder_attention_heads,
											)

											a_     :      Dict 						=     torch.cat(
											    [common_inputs["decoder_attention_mask"], torch.ones(a_			,     a_      )]			,     dim=1      )

											a_     :      Optional[int] 						=     []
											# If the number of encoder and decoder layers are present in the model configuration, both are considered
											a_						,				a_     :      Tuple 						=     self.num_layers
											a_     :      str 						=     min(a_			,     a_      )
											a_     :      Dict 						=     max(a_			,     a_      ) - min_num_layers
											a_     :      Union[str, Any] 						=     "encoder" if num_encoder_layers > num_decoder_layers else "decoder"

											for _ in range(a_      ):
													common_inputs["past_key_values"].append(
													    (
													        torch.zeros(a_      ),
													        torch.zeros(a_      ),
													        torch.zeros(a_      ),
													        torch.zeros(a_      ),
													    )      )
											# TODO: test this.
											a_     :      Optional[int] 						=     encoder_shape if remaining_side_name == "encoder" else decoder_shape
											for _ in range(a_			,     a_      ):
													common_inputs["past_key_values"].append((torch.zeros(a_      ), torch.zeros(a_      ))      )
									return common_inputs






							def snake_case_  (   self			,     a_			,     a_ = -1			,     a_ = -1			,     a_ = False			,     a_ = None			,     ):
									a_     :      Optional[int] 						=     self._generate_dummy_inputs_for_sequence_classification_and_question_answering(
									    a_			,     a_			,     a_			,     a_			,     a_      )

									if self.use_past:
											if not is_torch_available():
													raise ValueError("Cannot generate dummy past_keys inputs without PyTorch installed."      )
											else:
													import torch
											a_						,				a_     :      Dict 						=     common_inputs["input_ids"].shape
											# Not using the same length for past_key_values
											a_     :      int 						=     seqlen + 2
											a_						,				a_     :      Optional[int] 						=     self.num_layers
											a_						,				a_     :      Optional[int] 						=     self.num_attention_heads
											a_     :      Optional[Any] 						=     (
											    batch,
											    num_encoder_attention_heads,
											    past_key_values_length,
											    self._config.hidden_size // num_encoder_attention_heads,
											)

											a_     :      str 						=     common_inputs["attention_mask"].dtype
											a_     :      Tuple 						=     torch.cat(
											    [common_inputs["attention_mask"], torch.ones(a_			,     a_			,     dtype=a_      )]			,     dim=1      )
											a_     :      Optional[int] 						=     [
											    (torch.zeros(a_      ), torch.zeros(a_      )) for _ in range(a_      )
											]
									return common_inputs






							def snake_case_  (   self			,     a_			,     a_ = -1			,     a_ = -1			,     a_ = False			,     a_ = None			,     ):
									# Copied from OnnxConfig.generate_dummy_inputs
									# Did not use super(OnnxConfigWithPast, self).generate_dummy_inputs for code clarity.
									# If dynamic axis (-1) we forward with a fixed dimension of 2 samples to avoid optimizations made by ONNX
									a_     :      Optional[Any] 						=     compute_effective_axis_dimension(
									    a_			,     fixed_dimension=OnnxConfig.default_fixed_batch			,     num_token_to_add=0      )

									# If dynamic axis (-1) we forward with a fixed dimension of 8 tokens to avoid optimizations made by ONNX
									a_     :      Tuple 						=     tokenizer.num_special_tokens_to_add(a_      )
									a_     :      Union[str, Any] 						=     compute_effective_axis_dimension(
									    a_			,     fixed_dimension=OnnxConfig.default_fixed_sequence			,     num_token_to_add=a_      )

									# Generate dummy inputs according to compute batch and sequence
									a_     :      Union[str, Any] 						=     [" ".join([tokenizer.unk_token]      ) * seq_length] * batch_size
									a_     :      str 						=     dict(tokenizer(a_			,     return_tensors=a_      )      )
									return common_inputs






							def snake_case_  (   self			,     a_			,     a_ = -1			,     a_ = -1			,     a_ = False			,     a_ = None			,     ):
									if self.task in ["default", "seq2seq-lm"]:
											a_     :      List[str] 						=     self._generate_dummy_inputs_for_default_and_seqaseq_lm(
											    a_			,     batch_size=a_			,     seq_length=a_			,     is_pair=a_			,     framework=a_      )

									elif self.task == "causal-lm":
											a_     :      List[Any] 						=     self._generate_dummy_inputs_for_causal_lm(
											    a_			,     batch_size=a_			,     seq_length=a_			,     is_pair=a_			,     framework=a_      )
									else:
											a_     :      Union[str, Any] 						=     self._generate_dummy_inputs_for_sequence_classification_and_question_answering(
											    a_			,     batch_size=a_			,     seq_length=a_			,     is_pair=a_			,     framework=a_      )

									return common_inputs






							def snake_case_  (   self			,     a_			,     a_			,     a_			,     a_      ):
									if self.task in ["default", "seq2seq-lm"]:
											a_     :      Optional[int] 						=     super()._flatten_past_key_values_(a_			,     a_			,     a_			,     a_      )
									else:
											a_     :      int 						=     super(a_			,     self      )._flatten_past_key_values_(
											    a_			,     a_			,     a_			,     a_      )