


'''simple docstring'''
import pickle
import shutil
import tempfile
import unittest

from transformers import SPIECE_UNDERLINE, XGLMTokenizer, XGLMTokenizerFast
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_tokenizers, slow
from transformers.utils import cached_property

from ...test_tokenization_common import TokenizerTesterMixin


__snake_case							   =   get_tests_dir('''fixtures/test_sentencepiece.model''')
@require_sentencepiece
@require_tokenizers
class      lowercase			(							A__			,	unittest.TestCase      ):

		"""simple docstring"""

		_a									=       XGLMTokenizer
		_a									=       XGLMTokenizerFast
		_a									=       True
		_a									=       True




		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							super().setUp()

							# We have a SentencePiece fixture for testing
							UpperCamelCase__     :List[str]							      =				XGLMTokenizer(UpperCamelCase_					,     keep_accents=UpperCamelCase_							)
							tokenizer.save_pretrained(self.tmpdirname							)




		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							UpperCamelCase__     :List[Any]							      =				'''<pad>'''
							UpperCamelCase__     :List[Any]							      =				1

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




		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							UpperCamelCase__     :Optional[int]							      =				list(self.get_tokenizer().get_vocab().keys()							)

							self.assertEqual(vocab_keys[0]					,     '''<s>'''							)
							self.assertEqual(vocab_keys[1]					,     '''<pad>'''							)
							self.assertEqual(len(UpperCamelCase_							)					,     1008							)




		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							self.assertEqual(self.get_tokenizer().vocab_size					,     1008							)




		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							UpperCamelCase__     :int							      =				XGLMTokenizer(UpperCamelCase_					,     keep_accents=UpperCamelCase_							)

							UpperCamelCase__     :Tuple							      =				tokenizer.tokenize('''This is a test'''							)
							self.assertListEqual(UpperCamelCase_					,     ['''▁This''', '''▁is''', '''▁a''', '''▁t''', '''est''']							)

							self.assertListEqual(
							    tokenizer.convert_tokens_to_ids(UpperCamelCase_							)					,     [value + tokenizer.fairseq_offset for value in [285, 46, 10, 170, 382]]					,     )

							UpperCamelCase__     :Optional[Any]							      =				tokenizer.tokenize('''I was born in 92000, and this is falsé.'''							)
							self.assertListEqual(
							    UpperCamelCase_					,     [
							        SPIECE_UNDERLINE + '''I''',
							        SPIECE_UNDERLINE + '''was''',
							        SPIECE_UNDERLINE + '''b''',
							        '''or''',
							        '''n''',
							        SPIECE_UNDERLINE + '''in''',
							        SPIECE_UNDERLINE + '''''',
							        '''9''',
							        '''2''',
							        '''0''',
							        '''0''',
							        '''0''',
							        ''',''',
							        SPIECE_UNDERLINE + '''and''',
							        SPIECE_UNDERLINE + '''this''',
							        SPIECE_UNDERLINE + '''is''',
							        SPIECE_UNDERLINE + '''f''',
							        '''al''',
							        '''s''',
							        '''é''',
							        '''.''',
							    ]					,     )
							UpperCamelCase__     :Optional[Any]							      =				tokenizer.convert_tokens_to_ids(UpperCamelCase_							)
							self.assertListEqual(
							    UpperCamelCase_					,     [
							        value + tokenizer.fairseq_offset
							        for value in [8, 21, 84, 55, 24, 19, 7, 2, 602, 347, 347, 347, 3, 12, 66, 46, 72, 80, 6, 2, 4]
							    ]					,     )

							UpperCamelCase__     :List[str]							      =				tokenizer.convert_ids_to_tokens(UpperCamelCase_							)
							self.assertListEqual(
							    UpperCamelCase_					,     [
							        SPIECE_UNDERLINE + '''I''',
							        SPIECE_UNDERLINE + '''was''',
							        SPIECE_UNDERLINE + '''b''',
							        '''or''',
							        '''n''',
							        SPIECE_UNDERLINE + '''in''',
							        SPIECE_UNDERLINE + '''''',
							        '''<unk>''',
							        '''2''',
							        '''0''',
							        '''0''',
							        '''0''',
							        ''',''',
							        SPIECE_UNDERLINE + '''and''',
							        SPIECE_UNDERLINE + '''this''',
							        SPIECE_UNDERLINE + '''is''',
							        SPIECE_UNDERLINE + '''f''',
							        '''al''',
							        '''s''',
							        '''<unk>''',
							        '''.''',
							    ]					,     )




		@cached_property
		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							return XGLMTokenizer.from_pretrained('''facebook/xglm-564M'''							)




		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							with tempfile.NamedTemporaryFile() as f:
												shutil.copyfile(UpperCamelCase_					,     f.name							)
												UpperCamelCase__     :int							      =				XGLMTokenizer(f.name					,     keep_accents=UpperCamelCase_							)
												UpperCamelCase__     :Optional[Any]							      =				pickle.dumps(UpperCamelCase_							)
							pickle.loads(UpperCamelCase_							)




		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							if not self.test_rust_tokenizer:
												return

							UpperCamelCase__     :List[Any]							      =				self.get_tokenizer()
							UpperCamelCase__     :Optional[Any]							      =				self.get_rust_tokenizer()

							UpperCamelCase__     :Dict							      =				'''I was born in 92000, and this is falsé.'''

							UpperCamelCase__     :List[Any]							      =				tokenizer.tokenize(UpperCamelCase_							)
							UpperCamelCase__     :List[Any]							      =				rust_tokenizer.tokenize(UpperCamelCase_							)
							self.assertListEqual(UpperCamelCase_					,     UpperCamelCase_							)

							UpperCamelCase__     :Optional[Any]							      =				tokenizer.encode(UpperCamelCase_					,     add_special_tokens=UpperCamelCase_							)
							UpperCamelCase__     :Union[str, Any]							      =				rust_tokenizer.encode(UpperCamelCase_					,     add_special_tokens=UpperCamelCase_							)
							self.assertListEqual(UpperCamelCase_					,     UpperCamelCase_							)

							UpperCamelCase__     :Tuple							      =				self.get_rust_tokenizer()
							UpperCamelCase__     :Union[str, Any]							      =				tokenizer.encode(UpperCamelCase_							)
							UpperCamelCase__     :Union[str, Any]							      =				rust_tokenizer.encode(UpperCamelCase_							)
							self.assertListEqual(UpperCamelCase_					,     UpperCamelCase_							)




		@slow
		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							UpperCamelCase__     :Union[str, Any]							      =				'''Hello World!'''
							UpperCamelCase__     :Union[str, Any]							      =				[2, 31227, 4447, 35]

							self.assertListEqual(UpperCamelCase_					,     self.big_tokenizer.encode(UpperCamelCase_							)							)




		@slow
		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							UpperCamelCase__     :List[str]							      =				(
							    '''This is a very long text with a lot of weird characters, such as: . , ~ ? ( ) " [ ] ! : - . Also we will'''
							    ''' add words that should not exsist and be tokenized to unk, such as saoneuhaoesuth'''
							)
							# fmt: off
							UpperCamelCase__     :Any							      =				[2, 1018, 67, 11, 1988, 2617, 5631, 278, 11, 3407, 48, 71630, 28085, 4, 3234, 157, 13, 6, 5, 6, 4, 3526, 768, 15, 659, 57, 298, 3983, 864, 129, 21, 6, 5, 13675, 377, 652, 7580, 10341, 155, 2817, 422, 1666, 7, 1674, 53, 113, 202277, 17892, 33, 60, 87, 4, 3234, 157, 61, 2667, 52376, 19, 88, 23, 735]
							# fmt: on

							self.assertListEqual(UpperCamelCase_					,     self.big_tokenizer.encode(UpperCamelCase_							)							)




		@slow
		def 					lowerCAmelCase__       (	self							):

							'''simple docstring'''


							UpperCamelCase__     :List[Any]							      =				{
							    '''input_ids''': [[2, 108825, 1163, 15, 88010, 473, 15898, 157, 13672, 1857, 312, 8, 238021, 1163, 53, 13672, 1857, 312, 8, 53283, 182396, 8, 18566, 16, 36733, 4101, 8, 230, 244017, 122553, 7, 15, 132597, 4, 293, 12511, 7610, 4, 3414, 132597, 9, 4, 32361, 362, 4, 734, 28512, 32569, 18, 4, 32361, 26096, 14982, 73, 18715, 21433, 235261, 15, 492, 12427, 16, 53, 18715, 21433, 65454, 15, 23659, 563, 16, 278, 597, 2843, 595, 7931, 182396, 64186, 22, 886, 595, 132981, 53, 25540, 3449, 43982, 39901, 5951, 878, 330, 4, 27694, 80269, 312, 53, 6517, 11780, 611, 20408, 5], [2, 6, 132597, 67, 42897, 33, 592, 8, 163729, 25540, 361, 136997, 109514, 173230, 7, 501, 60, 102913, 196, 5631, 235, 63243, 473, 6, 231757, 74, 5277, 7905, 53, 3095, 37317, 22, 454, 183874, 5], [2, 268, 31298, 46530, 6, 132935, 43831, 7, 597, 32, 24, 3688, 9865, 5]],
							    '''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]]
							}  # noqa: E501
							# fmt: on

							self.tokenizer_integration_test_util(
							    expected_encoding=UpperCamelCase_					,     model_name='''facebook/xglm-564M'''					,     padding=UpperCamelCase_					,     )



'''simple docstring'''
import torch
from torch import nn
class      lowercase			(							nn.Module      ):

		"""simple docstring"""

		def __init__(	self					,     UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_=1					,     UpperCamelCase_=False							):

							'''simple docstring'''


							super().__init__()

							UpperCamelCase__     :Dict							      =				n_token
							UpperCamelCase__     :List[Any]							      =				d_embed
							UpperCamelCase__     :Dict							      =				d_proj

							UpperCamelCase__     :Dict							      =				cutoffs + [n_token]
							UpperCamelCase__     :Union[str, Any]							      =				[0] + self.cutoffs
							UpperCamelCase__     :Any							      =				div_val

							UpperCamelCase__     :int							      =				self.cutoffs[0]
							UpperCamelCase__     :List[Any]							      =				len(self.cutoffs							) - 1
							UpperCamelCase__     :List[Any]							      =				self.shortlist_size + self.n_clusters

							if self.n_clusters > 0:
												UpperCamelCase__     :Any							      =				nn.Parameter(torch.zeros(self.n_clusters					,     self.d_embed							)							)
												UpperCamelCase__     :Optional[Any]							      =				nn.Parameter(torch.zeros(self.n_clusters							)							)

							UpperCamelCase__     :Union[str, Any]							      =				nn.ModuleList()
							UpperCamelCase__     :str							      =				nn.ParameterList()

							if div_val == 1:
												for i in range(len(self.cutoffs							)							):
																	if d_proj != d_embed:
																						self.out_projs.append(nn.Parameter(torch.FloatTensor(UpperCamelCase_					,     UpperCamelCase_							)							)							)
																	else:
																						self.out_projs.append(UpperCamelCase_							)

												self.out_layers.append(nn.Linear(UpperCamelCase_					,     UpperCamelCase_							)							)
							else:
												for i in range(len(self.cutoffs							)							):
																	UpperCamelCase__      ,       UpperCamelCase__     :List[str]							      =				self.cutoff_ends[i], self.cutoff_ends[i + 1]
																	UpperCamelCase__     :Dict							      =				d_embed // (div_val**i)

																	self.out_projs.append(nn.Parameter(torch.FloatTensor(UpperCamelCase_					,     UpperCamelCase_							)							)							)

																	self.out_layers.append(nn.Linear(UpperCamelCase_					,     r_idx - l_idx							)							)

							UpperCamelCase__     :Tuple							      =				keep_order




		def 					lowerCAmelCase__       (	self					,     UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_							):

							'''simple docstring'''


							if proj is None:
												UpperCamelCase__     :List[str]							      =				nn.functional.linear(UpperCamelCase_					,     UpperCamelCase_					,     bias=UpperCamelCase_							)
							else:
												# if CUDA_MAJOR <= 9 and CUDA_MINOR <= 1:
												UpperCamelCase__     :Any							      =				nn.functional.linear(UpperCamelCase_					,     proj.t().contiguous()							)
												UpperCamelCase__     :Union[str, Any]							      =				nn.functional.linear(UpperCamelCase_					,     UpperCamelCase_					,     bias=UpperCamelCase_							)
												# else:
												#     logit = torch.einsum('bd,de,ev->bv', (hidden, proj, weight.t()))
												#     if bias is not None:
												#         logit = logit + bias

							return logit




		def 					lowerCAmelCase__       (	self					,     UpperCamelCase_					,     UpperCamelCase_=None					,     UpperCamelCase_=False							):

							'''simple docstring'''


							if labels is not None:
												# Shift so that tokens < n predict n
												UpperCamelCase__     :Optional[Any]							      =				hidden[..., :-1, :].contiguous()
												UpperCamelCase__     :Optional[Any]							      =				labels[..., 1:].contiguous()
												UpperCamelCase__     :Optional[Any]							      =				hidden.view(-1					,     hidden.size(-1							)							)
												UpperCamelCase__     :str							      =				labels.view(-1							)
												if hidden.size(0							) != labels.size(0							):
																	raise RuntimeError('''Input and labels should have the same size in the batch dimension.'''							)
							else:
												UpperCamelCase__     :int							      =				hidden.view(-1					,     hidden.size(-1							)							)

							if self.n_clusters == 0:
												UpperCamelCase__     :Optional[int]							      =				self._compute_logit(UpperCamelCase_					,     self.out_layers[0].weight					,     self.out_layers[0].bias					,     self.out_projs[0]							)
												if labels is not None:
																	UpperCamelCase__     :int							      =				labels != -100
																	UpperCamelCase__     :List[Any]							      =				torch.zeros_like(UpperCamelCase_					,     dtype=hidden.dtype					,     device=hidden.device							)
																	UpperCamelCase__     :str							      =				(
																	    -nn.functional.log_softmax(UpperCamelCase_					,     dim=-1							)[mask].gather(1					,     labels[mask].unsqueeze(1							)							).squeeze(1							)
																	)
												else:
																	UpperCamelCase__     :Dict							      =				nn.functional.log_softmax(UpperCamelCase_					,     dim=-1							)
							else:
												# construct weights and biases
												UpperCamelCase__      ,       UpperCamelCase__     :Dict							      =				[], []
												for i in range(len(self.cutoffs							)							):
																	if self.div_val == 1:
																						UpperCamelCase__      ,       UpperCamelCase__     :Union[str, Any]							      =				self.cutoff_ends[i], self.cutoff_ends[i + 1]
																						UpperCamelCase__     :str							      =				self.out_layers[0].weight[l_idx:r_idx]
																						UpperCamelCase__     :Optional[int]							      =				self.out_layers[0].bias[l_idx:r_idx]
																	else:
																						UpperCamelCase__     :Optional[Any]							      =				self.out_layers[i].weight
																						UpperCamelCase__     :Optional[int]							      =				self.out_layers[i].bias

																	if i == 0:
																						UpperCamelCase__     :str							      =				torch.cat([weight_i, self.cluster_weight]					,     dim=0							)
																						UpperCamelCase__     :List[str]							      =				torch.cat([bias_i, self.cluster_bias]					,     dim=0							)

																	weights.append(UpperCamelCase_							)
																	biases.append(UpperCamelCase_							)

												UpperCamelCase__      ,       UpperCamelCase__      ,       UpperCamelCase__     :Optional[Any]							      =				weights[0], biases[0], self.out_projs[0]

												UpperCamelCase__     :str							      =				self._compute_logit(UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_							)
												UpperCamelCase__     :List[Any]							      =				nn.functional.log_softmax(UpperCamelCase_					,     dim=1							)

												if labels is None:
																	UpperCamelCase__     :Union[str, Any]							      =				hidden.new_empty((head_logit.size(0							), self.n_token)							)
												else:
																	UpperCamelCase__     :Any							      =				torch.zeros_like(UpperCamelCase_					,     dtype=hidden.dtype					,     device=hidden.device							)

												UpperCamelCase__     :Any							      =				0
												UpperCamelCase__     :str							      =				[0] + self.cutoffs
												for i in range(len(UpperCamelCase_							) - 1							):
																	UpperCamelCase__      ,       UpperCamelCase__     :Optional[int]							      =				cutoff_values[i], cutoff_values[i + 1]

																	if labels is not None:
																						UpperCamelCase__     :Any							      =				(labels >= l_idx) & (labels < r_idx)
																						UpperCamelCase__     :int							      =				mask_i.nonzero().squeeze()

																						if indices_i.numel() == 0:
																											continue

																						UpperCamelCase__     :Tuple							      =				labels.index_select(0					,     UpperCamelCase_							) - l_idx
																						UpperCamelCase__     :str							      =				head_logprob.index_select(0					,     UpperCamelCase_							)
																						UpperCamelCase__     :int							      =				hidden.index_select(0					,     UpperCamelCase_							)
																	else:
																						UpperCamelCase__     :Dict							      =				hidden

																	if i == 0:
																						if labels is not None:
																											UpperCamelCase__     :Any							      =				head_logprob_i.gather(1					,     target_i[:, None]							).squeeze(1							)
																						else:
																											UpperCamelCase__     :List[str]							      =				head_logprob[:, : self.cutoffs[0]]
																	else:
																						UpperCamelCase__      ,       UpperCamelCase__      ,       UpperCamelCase__     :Dict							      =				weights[i], biases[i], self.out_projs[i]

																						UpperCamelCase__     :str							      =				self._compute_logit(UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_							)
																						UpperCamelCase__     :Union[str, Any]							      =				nn.functional.log_softmax(UpperCamelCase_					,     dim=1							)
																						UpperCamelCase__     :Any							      =				self.cutoffs[0] + i - 1  # No probability for the head cluster
																						if labels is not None:
																											UpperCamelCase__     :List[Any]							      =				head_logprob_i[:, cluster_prob_idx] + tail_logprob_i.gather(
																											    1					,     target_i[:, None]							).squeeze(1							)
																						else:
																											UpperCamelCase__     :List[str]							      =				head_logprob[:, cluster_prob_idx, None] + tail_logprob_i
																											UpperCamelCase__     :Dict							      =				logprob_i

																	if labels is not None:
																						if (hasattr(self					,     '''keep_order'''							) and self.keep_order) or keep_order:
																											out.index_copy_(0					,     UpperCamelCase_					,     -logprob_i							)
																						else:
																											out[offset : offset + logprob_i.size(0							)].copy_(-logprob_i							)
																						offset += logprob_i.size(0							)

							return out




		def 					lowerCAmelCase__       (	self					,     UpperCamelCase_							):

							'''simple docstring'''


							if self.n_clusters == 0:
												UpperCamelCase__     :Optional[Any]							      =				self._compute_logit(UpperCamelCase_					,     self.out_layers[0].weight					,     self.out_layers[0].bias					,     self.out_projs[0]							)
												return nn.functional.log_softmax(UpperCamelCase_					,     dim=-1							)
							else:
												# construct weights and biases
												UpperCamelCase__      ,       UpperCamelCase__     :List[Any]							      =				[], []
												for i in range(len(self.cutoffs							)							):
																	if self.div_val == 1:
																						UpperCamelCase__      ,       UpperCamelCase__     :Tuple							      =				self.cutoff_ends[i], self.cutoff_ends[i + 1]
																						UpperCamelCase__     :Union[str, Any]							      =				self.out_layers[0].weight[l_idx:r_idx]
																						UpperCamelCase__     :str							      =				self.out_layers[0].bias[l_idx:r_idx]
																	else:
																						UpperCamelCase__     :List[Any]							      =				self.out_layers[i].weight
																						UpperCamelCase__     :List[Any]							      =				self.out_layers[i].bias

																	if i == 0:
																						UpperCamelCase__     :Dict							      =				torch.cat([weight_i, self.cluster_weight]					,     dim=0							)
																						UpperCamelCase__     :Optional[Any]							      =				torch.cat([bias_i, self.cluster_bias]					,     dim=0							)

																	weights.append(UpperCamelCase_							)
																	biases.append(UpperCamelCase_							)

												UpperCamelCase__      ,       UpperCamelCase__      ,       UpperCamelCase__     :str							      =				weights[0], biases[0], self.out_projs[0]
												UpperCamelCase__     :Optional[int]							      =				self._compute_logit(UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_							)

												UpperCamelCase__     :int							      =				hidden.new_empty((head_logit.size(0							), self.n_token)							)
												UpperCamelCase__     :Union[str, Any]							      =				nn.functional.log_softmax(UpperCamelCase_					,     dim=1							)

												UpperCamelCase__     :int							      =				[0] + self.cutoffs
												for i in range(len(UpperCamelCase_							) - 1							):
																	UpperCamelCase__      ,       UpperCamelCase__     :Union[str, Any]							      =				cutoff_values[i], cutoff_values[i + 1]

																	if i == 0:
																						UpperCamelCase__     :str							      =				head_logprob[:, : self.cutoffs[0]]
																	else:
																						UpperCamelCase__      ,       UpperCamelCase__      ,       UpperCamelCase__     :Dict							      =				weights[i], biases[i], self.out_projs[i]

																						UpperCamelCase__     :List[str]							      =				self._compute_logit(UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_					,     UpperCamelCase_							)
																						UpperCamelCase__     :Tuple							      =				nn.functional.log_softmax(UpperCamelCase_					,     dim=1							)

																						UpperCamelCase__     :Optional[Any]							      =				head_logprob[:, -i] + tail_logprob_i
																						UpperCamelCase__     :Union[str, Any]							      =				logprob_i

												return out