


'''simple docstring'''



import unittest

from transformers import CamembertTokenizer, CamembertTokenizerFast
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_tokenizers, slow
from transformers.utils import is_torch_available

from ...test_tokenization_common import TokenizerTesterMixin


_lowercase      :      str            =					get_tests_dir("""fixtures/test_sentencepiece.model""")
_lowercase      :      Optional[Any]            =					get_tests_dir("""fixtures/test_sentencepiece_bpe.model""")

_lowercase      :      str            =					"""pt""" if is_torch_available() else """tf"""






@require_sentencepiece
@require_tokenizers
class 	UpperCamelCase__(       lowerCAmelCase						,    unittest.TestCase	):
				__magic_name__    :				Dict  		=	CamembertTokenizer
				__magic_name__    :				Optional[int]  		=	CamembertTokenizerFast
				__magic_name__    :				str  		=	True
				__magic_name__    :				Dict  		=	True




				def 					a__(		self      :			Tuple				)->					List[Any]:

										"""simple docstring"""




										super().setUp()

										# We have a SentencePiece fixture for testing
										UpperCAmelCase					    =     CamembertTokenizer(lowerCAmelCase				)
										tokenizer.save_pretrained(self.tmpdirname				)




				def 					a__(		self      :			Union[str, Any]				)->					Optional[Any]:

										"""simple docstring"""




										UpperCAmelCase					    =     '''<pad>'''
										UpperCAmelCase					    =     1

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




				def 					a__(		self      :			Dict				)->					Tuple:

										"""simple docstring"""




										UpperCAmelCase					    =     list(self.get_tokenizer().get_vocab().keys()				)

										self.assertEqual(vocab_keys[0]				,   '''<s>NOTUSED'''				)
										self.assertEqual(vocab_keys[1]				,   '''<pad>'''				)
										self.assertEqual(vocab_keys[-1]				,   '''<mask>'''				)
										self.assertEqual(len(lowerCAmelCase				)				,   1004				)




				def 					a__(		self      :			Dict				)->					Optional[Any]:

										"""simple docstring"""




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




				def 					a__(		self      :			Dict				)->					Optional[int]:

										"""simple docstring"""




										UpperCAmelCase					    =     CamembertTokenizer(lowerCAmelCase				)
										tokenizer.save_pretrained(self.tmpdirname				)
										UpperCAmelCase					    =     CamembertTokenizerFast.from_pretrained(self.tmpdirname				)

										UpperCAmelCase					    =     '''I was born in 92000, and this is falsé.'''

										UpperCAmelCase					    =     tokenizer.encode(lowerCAmelCase				)
										UpperCAmelCase					    =     rust_tokenizer.encode(lowerCAmelCase				)
										self.assertListEqual(lowerCAmelCase				,   lowerCAmelCase				)

										UpperCAmelCase					    =     tokenizer.encode(lowerCAmelCase				,   add_special_tokens=lowerCAmelCase				)
										UpperCAmelCase					    =     rust_tokenizer.encode(lowerCAmelCase				,   add_special_tokens=lowerCAmelCase				)
										self.assertListEqual(lowerCAmelCase				,   lowerCAmelCase				)

										# <unk> tokens are not the same for `rust` than for `slow`.
										# Because spm gives back raw token instead of `unk` in EncodeAsPieces
										# tokens = tokenizer.tokenize(sequence)
										UpperCAmelCase					    =     tokenizer.convert_ids_to_tokens(lowerCAmelCase				)
										UpperCAmelCase					    =     rust_tokenizer.tokenize(lowerCAmelCase				)
										self.assertListEqual(lowerCAmelCase				,   lowerCAmelCase				)




				def 					a__(		self      :			List[str]				)->					str:

										"""simple docstring"""




										if not self.test_rust_tokenizer:
																return

										UpperCAmelCase					    =     self.get_tokenizer()
										UpperCAmelCase					    =     self.get_rust_tokenizer()

										UpperCAmelCase					    =     '''I was born in 92000, and this is falsé.'''

										UpperCAmelCase					    =     tokenizer.tokenize(lowerCAmelCase				)
										UpperCAmelCase					    =     rust_tokenizer.tokenize(lowerCAmelCase				)
										self.assertListEqual(lowerCAmelCase				,   lowerCAmelCase				)

										UpperCAmelCase					    =     tokenizer.encode(lowerCAmelCase				,   add_special_tokens=lowerCAmelCase				)
										UpperCAmelCase					    =     rust_tokenizer.encode(lowerCAmelCase				,   add_special_tokens=lowerCAmelCase				)
										self.assertListEqual(lowerCAmelCase				,   lowerCAmelCase				)

										UpperCAmelCase					    =     self.get_rust_tokenizer()
										UpperCAmelCase					    =     tokenizer.encode(lowerCAmelCase				)
										UpperCAmelCase					    =     rust_tokenizer.encode(lowerCAmelCase				)
										self.assertListEqual(lowerCAmelCase				,   lowerCAmelCase				)





				@slow
				def 					a__(		self      :			Union[str, Any]				)->					List[Any]:

										"""simple docstring"""




										UpperCAmelCase					    =     {'''input_ids''': [[5, 54, 7196, 297, 30, 23, 776, 18, 11, 3215, 3705, 8252, 22, 3164, 1181, 2116, 29, 16, 813, 25, 791, 3314, 20, 3446, 38, 27575, 120, 6, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], [5, 468, 17, 11, 9088, 20, 1517, 8, 22804, 18818, 10, 38, 629, 607, 607, 142, 19, 7196, 867, 56, 10326, 24, 2267, 20, 416, 5072, 15612, 233, 734, 7, 2399, 27, 16, 3015, 1649, 7, 24, 20, 4338, 2399, 27, 13, 3400, 14, 13, 6189, 8, 930, 9, 6]], '''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, 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, 1, 1, 1, 1]]}  # noqa: E501
										# fmt: on

										# camembert is a french model. So we also use french texts.
										UpperCAmelCase					    =     [
										    '''Le transformeur est un modèle d\'apprentissage profond introduit en 2017, '''
										    '''utilisé principalement dans le domaine du traitement automatique des langues (TAL).''',
										    '''À l\'instar des réseaux de neurones récurrents (RNN), les transformeurs sont conçus '''
										    '''pour gérer des données séquentielles, telles que le langage naturel, pour des tâches '''
										    '''telles que la traduction et la synthèse de texte.''',
										]

										self.tokenizer_integration_test_util(
										    expected_encoding=lowerCAmelCase				,   model_name='''camembert-base'''				,   revision='''3a0641d9a1aeb7e848a74299e7e4c4bca216b4cf'''				,   sequences=lowerCAmelCase				,   )









'''simple docstring'''



import copy
import inspect
import unittest

from transformers import AutoBackbone
from transformers.configuration_utils import PretrainedConfig
from transformers.testing_utils import require_timm, require_torch, torch_device
from transformers.utils.import_utils import is_torch_available

from ...test_backbone_common import BackboneTesterMixin
from ...test_configuration_common import ConfigTester
from ...test_modeling_common import ModelTesterMixin, floats_tensor


if is_torch_available():
	import torch

	from transformers import TimmBackbone, TimmBackboneConfig

from ...test_pipeline_mixin import PipelineTesterMixin






class 	UpperCamelCase__:




				def __init__(		self      :			List[str]				,   lowerCAmelCase      :			int				,   lowerCAmelCase      :			Optional[int]=None				,   lowerCAmelCase      :			Optional[int]=None				,   lowerCAmelCase      :			List[str]=None				,   lowerCAmelCase      :			int="resnet50"				,   lowerCAmelCase      :			Optional[int]=3				,   lowerCAmelCase      :			Union[str, Any]=32				,   lowerCAmelCase      :			List[Any]=3				,   lowerCAmelCase      :			Tuple=True				,   lowerCAmelCase      :			str=True				,   )->					Union[str, Any]:

										"""simple docstring"""




										UpperCAmelCase					    =     parent
										UpperCAmelCase					    =     out_indices if out_indices is not None else [4]
										UpperCAmelCase					    =     stage_names
										UpperCAmelCase					    =     out_features
										UpperCAmelCase					    =     backbone
										UpperCAmelCase					    =     batch_size
										UpperCAmelCase					    =     image_size
										UpperCAmelCase					    =     num_channels
										UpperCAmelCase					    =     use_pretrained_backbone
										UpperCAmelCase					    =     is_training




				def 					a__(		self      :			Optional[Any]				)->					Optional[int]:

										"""simple docstring"""




										UpperCAmelCase					    =     floats_tensor([self.batch_size, self.num_channels, self.image_size, self.image_size]				)
										UpperCAmelCase					    =     self.get_config()

										return config, pixel_values




				def 					a__(		self      :			int				)->					Optional[Any]:

										"""simple docstring"""




										return TimmBackboneConfig(
										    image_size=self.image_size				,   num_channels=self.num_channels				,   out_features=self.out_features				,   out_indices=self.out_indices				,   stage_names=self.stage_names				,   use_pretrained_backbone=self.use_pretrained_backbone				,   backbone=self.backbone				,   )




				def 					a__(		self      :			Dict				,   lowerCAmelCase      :			str				,   lowerCAmelCase      :			Dict				)->					Tuple:

										"""simple docstring"""




										UpperCAmelCase					    =     TimmBackbone(config=lowerCAmelCase				)
										model.to(lowerCAmelCase				)
										model.eval()
										with torch.no_grad():
																UpperCAmelCase					    =     model(lowerCAmelCase				)
										self.parent.assertEqual(
										    result.feature_map[-1].shape				,   (self.batch_size, model.channels[-1], 14, 14)				,   )





				def 					a__(		self      :			str				)->					Optional[Any]:

										"""simple docstring"""




										UpperCAmelCase					    =     self.prepare_config_and_inputs()
										UpperCAmelCase    ,    UpperCAmelCase					    =     config_and_inputs
										UpperCAmelCase					    =     {'''pixel_values''': pixel_values}
										return config, inputs_dict







@require_torch
@require_timm
class 	UpperCamelCase__(       lowerCAmelCase						,    lowerCAmelCase						,    lowerCAmelCase						,    unittest.TestCase	):
				__magic_name__    :				List[str]  		=	(TimmBackbone,) if is_torch_available() else ()
				__magic_name__    :				Any  		=	{"feature-extraction": TimmBackbone} if is_torch_available() else {}
				__magic_name__    :				Union[str, Any]  		=	False
				__magic_name__    :				int  		=	False
				__magic_name__    :				Tuple  		=	False
				__magic_name__    :				List[str]  		=	False




				def 					a__(		self      :			int				)->					str:

										"""simple docstring"""




										UpperCAmelCase					    =     TimmBackboneModelTester(self				)
										UpperCAmelCase					    =     ConfigTester(self				,   config_class=lowerCAmelCase				,   has_text_modality=lowerCAmelCase				)




				def 					a__(		self      :			List[str]				)->					Dict:

										"""simple docstring"""




										self.config_tester.create_and_test_config_to_json_string()
										self.config_tester.create_and_test_config_to_json_file()
										self.config_tester.create_and_test_config_from_and_save_pretrained()
										self.config_tester.create_and_test_config_with_num_labels()
										self.config_tester.check_config_can_be_init_without_params()
										self.config_tester.check_config_arguments_init()




				def 					a__(		self      :			List[Any]				)->					Optional[Any]:

										"""simple docstring"""




										UpperCAmelCase					    =     '''resnet18'''
										UpperCAmelCase					    =     '''microsoft/resnet-18'''

										UpperCAmelCase					    =     AutoBackbone.from_pretrained(lowerCAmelCase				,   use_timm_backbone=lowerCAmelCase				)
										UpperCAmelCase					    =     AutoBackbone.from_pretrained(lowerCAmelCase				)

										self.assertEqual(len(timm_model.out_features				)				,   len(transformers_model.out_features				)				)
										self.assertEqual(len(timm_model.stage_names				)				,   len(transformers_model.stage_names				)				)
										self.assertEqual(timm_model.channels				,   transformers_model.channels				)
										# Out indices are set to the last layer by default. For timm models, we don't know
										# the number of layers in advance, so we set it to (-1,), whereas for transformers
										# models, we set it to [len(stage_names) - 1] (kept for backward compatibility).
										self.assertEqual(timm_model.out_indices				,   (-1,)				)
										self.assertEqual(transformers_model.out_indices				,   [len(timm_model.stage_names				) - 1]				)

										UpperCAmelCase					    =     AutoBackbone.from_pretrained(lowerCAmelCase				,   use_timm_backbone=lowerCAmelCase				,   out_indices=[1, 2, 3]				)
										UpperCAmelCase					    =     AutoBackbone.from_pretrained(lowerCAmelCase				,   out_indices=[1, 2, 3]				)

										self.assertEqual(timm_model.out_indices				,   transformers_model.out_indices				)
										self.assertEqual(len(timm_model.out_features				)				,   len(transformers_model.out_features				)				)
										self.assertEqual(timm_model.channels				,   transformers_model.channels				)




				@unittest.skip('''TimmBackbone doesn\'t support feed forward chunking'''				)
				def 					a__(		self      :			Union[str, Any]				)->					Tuple:

										"""simple docstring"""




										pass




				@unittest.skip('''TimmBackbone doesn\'t have num_hidden_layers attribute'''				)
				def 					a__(		self      :			Dict				)->					Dict:

										"""simple docstring"""




										pass




				@unittest.skip('''TimmBackbone initialization is managed on the timm side'''				)
				def 					a__(		self      :			Tuple				)->					List[Any]:

										"""simple docstring"""




										pass




				@unittest.skip('''TimmBackbone models doesn\'t have inputs_embeds'''				)
				def 					a__(		self      :			List[str]				)->					Optional[Any]:

										"""simple docstring"""




										pass




				@unittest.skip('''TimmBackbone models doesn\'t have inputs_embeds'''				)
				def 					a__(		self      :			Optional[Any]				)->					Optional[Any]:

										"""simple docstring"""




										pass




				@unittest.skip('''TimmBackbone model cannot be created without specifying a backbone checkpoint'''				)
				def 					a__(		self      :			Any				)->					Optional[int]:

										"""simple docstring"""




										pass




				@unittest.skip('''Only checkpoints on timm can be loaded into TimmBackbone'''				)
				def 					a__(		self      :			Optional[int]				)->					Optional[int]:

										"""simple docstring"""




										pass




				@unittest.skip('''model weights aren\'t tied in TimmBackbone.'''				)
				def 					a__(		self      :			str				)->					str:

										"""simple docstring"""




										pass




				@unittest.skip('''model weights aren\'t tied in TimmBackbone.'''				)
				def 					a__(		self      :			Any				)->					int:

										"""simple docstring"""




										pass




				@unittest.skip('''Only checkpoints on timm can be loaded into TimmBackbone'''				)
				def 					a__(		self      :			List[Any]				)->					int:

										"""simple docstring"""




										pass




				@unittest.skip('''Only checkpoints on timm can be loaded into TimmBackbone'''				)
				def 					a__(		self      :			List[Any]				)->					List[str]:

										"""simple docstring"""




										pass




				@unittest.skip('''TimmBackbone doesn\'t have hidden size info in its configuration.'''				)
				def 					a__(		self      :			List[str]				)->					Dict:

										"""simple docstring"""




										pass




				@unittest.skip('''TimmBackbone doesn\'t support output_attentions.'''				)
				def 					a__(		self      :			str				)->					Tuple:

										"""simple docstring"""




										pass




				@unittest.skip('''Safetensors is not supported by timm.'''				)
				def 					a__(		self      :			List[Any]				)->					List[Any]:

										"""simple docstring"""




										pass




				@unittest.skip('''Will be fixed soon by reducing the size of the model used for common tests.'''				)
				def 					a__(		self      :			int				)->					Tuple:

										"""simple docstring"""




										pass




				def 					a__(		self      :			Dict				)->					Union[str, Any]:

										"""simple docstring"""




										UpperCAmelCase    ,    UpperCAmelCase					    =     self.model_tester.prepare_config_and_inputs_for_common()

										for model_class in self.all_model_classes:
																UpperCAmelCase					    =     model_class(lowerCAmelCase				)
																UpperCAmelCase					    =     inspect.signature(model.forward				)
																# signature.parameters is an OrderedDict => so arg_names order is deterministic
																UpperCAmelCase					    =     [*signature.parameters.keys()]

																UpperCAmelCase					    =     ['''pixel_values''']
																self.assertListEqual(arg_names[:1]				,   lowerCAmelCase				)




				def 					a__(		self      :			Union[str, Any]				)->					Optional[Any]:

										"""simple docstring"""




										UpperCAmelCase    ,    UpperCAmelCase					    =     self.model_tester.prepare_config_and_inputs_for_common()
										UpperCAmelCase					    =     True
										UpperCAmelCase					    =     self.has_attentions

										# no need to test all models as different heads yield the same functionality
										UpperCAmelCase					    =     self.all_model_classes[0]
										UpperCAmelCase					    =     model_class(lowerCAmelCase				)
										model.to(lowerCAmelCase				)

										UpperCAmelCase					    =     self._prepare_for_class(lowerCAmelCase				,   lowerCAmelCase				)
										UpperCAmelCase					    =     model(**lowerCAmelCase				)
										UpperCAmelCase					    =     outputs[0][-1]

										# Encoder-/Decoder-only models
										UpperCAmelCase					    =     outputs.hidden_states[0]
										hidden_states.retain_grad()

										if self.has_attentions:
																UpperCAmelCase					    =     outputs.attentions[0]
																attentions.retain_grad()

										output.flatten()[0].backward(retain_graph=lowerCAmelCase				)

										self.assertIsNotNone(hidden_states.grad				)

										if self.has_attentions:
																self.assertIsNotNone(attentions.grad				)





				def 					a__(		self      :			Optional[int]				)->					Optional[Any]:

										"""simple docstring"""




										UpperCAmelCase    ,    UpperCAmelCase					    =     self.model_tester.prepare_config_and_inputs_for_common()

										for model_class in self.all_model_classes:
																UpperCAmelCase					    =     model_class(lowerCAmelCase				)
																model.to(lowerCAmelCase				)
																model.eval()
																UpperCAmelCase					    =     model(**lowerCAmelCase				)

																self.assertEqual(len(result.feature_maps				)				,   len(config.out_indices				)				)
																self.assertEqual(len(model.channels				)				,   len(config.out_indices				)				)

																# Check output of last stage is taken if out_features=None, out_indices=None
																UpperCAmelCase					    =     copy.deepcopy(lowerCAmelCase				)
																UpperCAmelCase					    =     None
																UpperCAmelCase					    =     model_class(lowerCAmelCase				)
																model.to(lowerCAmelCase				)
																model.eval()
																UpperCAmelCase					    =     model(**lowerCAmelCase				)

																self.assertEqual(len(result.feature_maps				)				,   1				)
																self.assertEqual(len(model.channels				)				,   1				)

																# Check backbone can be initialized with fresh weights
																UpperCAmelCase					    =     copy.deepcopy(lowerCAmelCase				)
																UpperCAmelCase					    =     False
																UpperCAmelCase					    =     model_class(lowerCAmelCase				)
																model.to(lowerCAmelCase				)
																model.eval()
																UpperCAmelCase					    =     model(**lowerCAmelCase				)





