
import unittest

from transformers import MobileBertConfig, is_torch_available
from transformers.models.auto import get_values
from transformers.testing_utils import require_sentencepiece, require_tokenizers, require_torch, slow, torch_device

from ...test_configuration_common import ConfigTester
from ...test_modeling_common import ModelTesterMixin, ids_tensor, random_attention_mask
from ...test_pipeline_mixin import PipelineTesterMixin


if is_torch_available():
					import torch

					from transformers import (
					    MODEL_FOR_PRETRAINING_MAPPING,
					    MobileBertForMaskedLM,
					    MobileBertForMultipleChoice,
					    MobileBertForNextSentencePrediction,
					    MobileBertForPreTraining,
					    MobileBertForQuestionAnswering,
					    MobileBertForSequenceClassification,
					    MobileBertForTokenClassification,
					    MobileBertModel,
					)





class 							UpperCAmelCase      :




	'''simple docstring'''





	def __init__(       self					: int							,A					: Tuple							,A					: List[str]=13							,A					: List[str]=7							,A					: str=True							,A					: int=True							,A					: Union[str, Any]=True							,A					: str=True							,A					: Dict=99							,A					: Optional[Any]=64							,A					: Union[str, Any]=32							,A					: Dict=5							,A					: Union[str, Any]=4							,A					: str=37							,A					: int="gelu"							,A					: str=0.1							,A					: List[Any]=0.1							,A					: Union[str, Any]=5_12							,A					: Tuple=16							,A					: List[Any]=2							,A					: Any=0.02							,A					: List[Any]=3							,A					: Union[str, Any]=4							,A					: List[str]=None							,):
					__A   			= parent
					__A   			= batch_size
					__A   			= seq_length
					__A   			= is_training
					__A   			= use_input_mask
					__A   			= use_token_type_ids
					__A   			= use_labels
					__A   			= vocab_size
					__A   			= hidden_size
					__A   			= embedding_size
					__A   			= num_hidden_layers
					__A   			= num_attention_heads
					__A   			= intermediate_size
					__A   			= hidden_act
					__A   			= hidden_dropout_prob
					__A   			= attention_probs_dropout_prob
					__A   			= max_position_embeddings
					__A   			= type_vocab_size
					__A   			= type_sequence_label_size
					__A   			= initializer_range
					__A   			= num_labels
					__A   			= num_choices
					__A   			= scope






	def 				UpperCamelCase_  (       self					: List[Any] ):
					__A   			= ids_tensor([self.batch_size, self.seq_length]							,self.vocab_size )

					__A   			= None
					if self.use_input_mask:
									__A   			= random_attention_mask([self.batch_size, self.seq_length] )

					__A   			= None
					if self.use_token_type_ids:
									__A   			= ids_tensor([self.batch_size, self.seq_length]							,self.type_vocab_size )

					__A   			= None
					__A   			= None
					__A   			= None
					if self.use_labels:
									__A   			= ids_tensor([self.batch_size]							,self.type_sequence_label_size )
									__A   			= ids_tensor([self.batch_size, self.seq_length]							,self.num_labels )
									__A   			= ids_tensor([self.batch_size]							,self.num_choices )

					__A   			= self.get_config()

					return config, input_ids, token_type_ids, input_mask, sequence_labels, token_labels, choice_labels






	def 				UpperCamelCase_  (       self					: Any ):
					return MobileBertConfig(
					    vocab_size=self.vocab_size							,hidden_size=self.hidden_size							,num_hidden_layers=self.num_hidden_layers							,num_attention_heads=self.num_attention_heads							,intermediate_size=self.intermediate_size							,embedding_size=self.embedding_size							,hidden_act=self.hidden_act							,hidden_dropout_prob=self.hidden_dropout_prob							,attention_probs_dropout_prob=self.attention_probs_dropout_prob							,max_position_embeddings=self.max_position_embeddings							,type_vocab_size=self.type_vocab_size							,is_decoder=A							,initializer_range=self.initializer_range							,)






	def 				UpperCamelCase_  (       self					: Dict							,A					: Tuple							,A					: int							,A					: List[Any]							,A					: Optional[Any]							,A					: Optional[int]							,A					: int							,A					: int ):
					__A   			= MobileBertModel(config=A )
					model.to(A )
					model.eval()
					__A   			= model(A							,attention_mask=A							,token_type_ids=A )
					__A   			= model(A							,token_type_ids=A )
					__A   			= model(A )

					self.parent.assertEqual(result.last_hidden_state.shape							,(self.batch_size, self.seq_length, self.hidden_size) )
					self.parent.assertEqual(result.pooler_output.shape							,(self.batch_size, self.hidden_size) )






	def 				UpperCamelCase_  (       self					: int							,A					: Dict							,A					: int							,A					: List[Any]							,A					: Optional[Any]							,A					: Tuple							,A					: str							,A					: str ):
					__A   			= MobileBertForMaskedLM(config=A )
					model.to(A )
					model.eval()
					__A   			= model(A							,attention_mask=A							,token_type_ids=A							,labels=A )
					self.parent.assertEqual(result.logits.shape							,(self.batch_size, self.seq_length, self.vocab_size) )






	def 				UpperCamelCase_  (       self					: Any							,A					: Optional[Any]							,A					: Optional[Any]							,A					: Dict							,A					: Optional[int]							,A					: Any							,A					: Optional[Any]							,A					: Any ):
					__A   			= MobileBertForNextSentencePrediction(config=A )
					model.to(A )
					model.eval()
					__A   			= model(
					    A							,attention_mask=A							,token_type_ids=A							,labels=A							,)
					self.parent.assertEqual(result.logits.shape							,(self.batch_size, 2) )






	def 				UpperCamelCase_  (       self					: Dict							,A					: int							,A					: str							,A					: Optional[Any]							,A					: Optional[int]							,A					: Dict							,A					: Tuple							,A					: Optional[int] ):
					__A   			= MobileBertForPreTraining(config=A )
					model.to(A )
					model.eval()
					__A   			= model(
					    A							,attention_mask=A							,token_type_ids=A							,labels=A							,next_sentence_label=A							,)
					self.parent.assertEqual(result.prediction_logits.shape							,(self.batch_size, self.seq_length, self.vocab_size) )
					self.parent.assertEqual(result.seq_relationship_logits.shape							,(self.batch_size, 2) )






	def 				UpperCamelCase_  (       self					: int							,A					: Tuple							,A					: Any							,A					: Dict							,A					: List[Any]							,A					: Union[str, Any]							,A					: Optional[int]							,A					: Optional[int] ):
					__A   			= MobileBertForQuestionAnswering(config=A )
					model.to(A )
					model.eval()
					__A   			= model(
					    A							,attention_mask=A							,token_type_ids=A							,start_positions=A							,end_positions=A							,)
					self.parent.assertEqual(result.start_logits.shape							,(self.batch_size, self.seq_length) )
					self.parent.assertEqual(result.end_logits.shape							,(self.batch_size, self.seq_length) )






	def 				UpperCamelCase_  (       self					: Tuple							,A					: List[str]							,A					: List[str]							,A					: Optional[Any]							,A					: List[Any]							,A					: List[str]							,A					: Dict							,A					: Tuple ):
					__A   			= self.num_labels
					__A   			= MobileBertForSequenceClassification(A )
					model.to(A )
					model.eval()
					__A   			= model(A							,attention_mask=A							,token_type_ids=A							,labels=A )
					self.parent.assertEqual(result.logits.shape							,(self.batch_size, self.num_labels) )






	def 				UpperCamelCase_  (       self					: Optional[int]							,A					: Dict							,A					: Optional[int]							,A					: str							,A					: Dict							,A					: Union[str, Any]							,A					: int							,A					: Union[str, Any] ):
					__A   			= self.num_labels
					__A   			= MobileBertForTokenClassification(config=A )
					model.to(A )
					model.eval()
					__A   			= model(A							,attention_mask=A							,token_type_ids=A							,labels=A )
					self.parent.assertEqual(result.logits.shape							,(self.batch_size, self.seq_length, self.num_labels) )






	def 				UpperCamelCase_  (       self					: Optional[int]							,A					: int							,A					: str							,A					: Optional[int]							,A					: List[Any]							,A					: Optional[Any]							,A					: Optional[Any]							,A					: Optional[Any] ):
					__A   			= self.num_choices
					__A   			= MobileBertForMultipleChoice(config=A )
					model.to(A )
					model.eval()
					__A   			= input_ids.unsqueeze(1 ).expand(-1							,self.num_choices							,-1 ).contiguous()
					__A   			= token_type_ids.unsqueeze(1 ).expand(-1							,self.num_choices							,-1 ).contiguous()
					__A   			= input_mask.unsqueeze(1 ).expand(-1							,self.num_choices							,-1 ).contiguous()
					__A   			= model(
					    A							,attention_mask=A							,token_type_ids=A							,labels=A							,)
					self.parent.assertEqual(result.logits.shape							,(self.batch_size, self.num_choices) )







	def 				UpperCamelCase_  (       self					: Tuple ):
					__A   			= self.prepare_config_and_inputs()
					(
					    (
					    __A
					)			,							(
					    __A
					)			,							(
					    __A
					)			,							(
					    __A
					)			,							(
					    __A
					)			,							(
					    __A
					)			,							(
					    __A
					)			,							
					)   			= config_and_inputs
					__A   			= {"input_ids": input_ids, "token_type_ids": token_type_ids, "attention_mask": input_mask}
					return config, inputs_dict





@require_torch
class 							UpperCAmelCase      (  __SCREAMING_SNAKE_CASE			,			__SCREAMING_SNAKE_CASE			,			unittest.TestCase  ):




	'''simple docstring'''





	snake_case_					    =     (
	    (
	        MobileBertModel,
	        MobileBertForMaskedLM,
	        MobileBertForMultipleChoice,
	        MobileBertForNextSentencePrediction,
	        MobileBertForPreTraining,
	        MobileBertForQuestionAnswering,
	        MobileBertForSequenceClassification,
	        MobileBertForTokenClassification,
	    )
	    if is_torch_available()
	    else ()
	)
	snake_case_					    =     (
	    {
	        "feature-extraction": MobileBertModel,
	        "fill-mask": MobileBertForMaskedLM,
	        "question-answering": MobileBertForQuestionAnswering,
	        "text-classification": MobileBertForSequenceClassification,
	        "token-classification": MobileBertForTokenClassification,
	        "zero-shot": MobileBertForSequenceClassification,
	    }
	    if is_torch_available()
	    else {}
	)
	snake_case_					    =     True






	def 				UpperCamelCase_  (       self					: List[str]							,A					: Tuple							,A					: List[Any]							,A					: Union[str, Any]=False ):
					__A   			= super()._prepare_for_class(A							,A							,return_labels=A )

					if return_labels:
									if model_class in get_values(A ):
													__A   			= torch.zeros(
													    (self.model_tester.batch_size, self.model_tester.seq_length)							,dtype=torch.long							,device=A )
													__A   			= torch.zeros(
													    self.model_tester.batch_size							,dtype=torch.long							,device=A )
					return inputs_dict






	def 				UpperCamelCase_  (       self					: int ):
					__A   			= MobileBertModelTester(self )
					__A   			= ConfigTester(self							,config_class=A							,hidden_size=37 )






	def 				UpperCamelCase_  (       self					: Any ):
					self.config_tester.run_common_tests()






	def 				UpperCamelCase_  (       self					: Any ):
					__A   			= self.model_tester.prepare_config_and_inputs()
					self.model_tester.create_and_check_mobilebert_model(*A )






	def 				UpperCamelCase_  (       self					: Union[str, Any] ):
					__A   			= self.model_tester.prepare_config_and_inputs()
					self.model_tester.create_and_check_mobilebert_for_masked_lm(*A )






	def 				UpperCamelCase_  (       self					: Optional[Any] ):
					__A   			= self.model_tester.prepare_config_and_inputs()
					self.model_tester.create_and_check_mobilebert_for_multiple_choice(*A )






	def 				UpperCamelCase_  (       self					: Optional[Any] ):
					__A   			= self.model_tester.prepare_config_and_inputs()
					self.model_tester.create_and_check_mobilebert_for_next_sequence_prediction(*A )






	def 				UpperCamelCase_  (       self					: Optional[int] ):
					__A   			= self.model_tester.prepare_config_and_inputs()
					self.model_tester.create_and_check_mobilebert_for_pretraining(*A )






	def 				UpperCamelCase_  (       self					: Any ):
					__A   			= self.model_tester.prepare_config_and_inputs()
					self.model_tester.create_and_check_mobilebert_for_question_answering(*A )






	def 				UpperCamelCase_  (       self					: str ):
					__A   			= self.model_tester.prepare_config_and_inputs()
					self.model_tester.create_and_check_mobilebert_for_sequence_classification(*A )







	def 				UpperCamelCase_  (       self					: int ):
					__A   			= self.model_tester.prepare_config_and_inputs()
					self.model_tester.create_and_check_mobilebert_for_token_classification(*A )





def 						UpperCAmelCase			(	a_						)		->	str:







				"""simple docstring"""



				return torch.tensor(
				    a_		,   dtype=torch.long		,   device=a_		,   )


SCREAMING_SNAKE_CASE							:List[str]							 =				1E-3





@require_torch
@require_sentencepiece
@require_tokenizers
class 							UpperCAmelCase      (  unittest.TestCase  ):




	'''simple docstring'''





	@slow
	def 				UpperCamelCase_  (       self					: Tuple ):
					__A   			= MobileBertModel.from_pretrained("google/mobilebert-uncased" ).to(A )
					__A   			= _long_tensor([[1_01, 71_10, 10_05, 10_56, 20_23, 1_13_33, 1_74_13, 10_29, 1_02]] )
					with torch.no_grad():
									__A   			= model(A )[0]
					__A   			= torch.Size((1, 9, 5_12) )
					self.assertEqual(output.shape							,A )
					__A   			= torch.tensor(
					    [
					        [
					            [-2.4736526E07, 8.2691656E04, 1.6521838E05],
					            [-5.7541704E-01, 3.9056022E00, 4.4011507E00],
					            [2.6047359E00, 1.5677652E00, -1.7324188E-01],
					        ]
					    ]							,device=A							,)

					# MobileBERT results range from 10e0 to 10e8. Even a 0.0000001% difference with a value of 10e8 results in a
					# ~1 difference, it's therefore not a good idea to measure using addition.
					# Here, we instead divide the expected result with the result in order to obtain ~1. We then check that the
					# result is held between bounds: 1 - TOLERANCE < expected_result / result < 1 + TOLERANCE
					__A   			= torch.all((expected_slice / output[..., :3, :3]) >= 1 - TOLERANCE )
					__A   			= torch.all((expected_slice / output[..., :3, :3]) <= 1 + TOLERANCE )

					self.assertTrue(lower_bound and upper_bound )







'''simple docstring'''


from ...configuration_utils import PretrainedConfig
from ...utils import logging


lowerCAmelCase					:     Optional[Any]												=logging.get_logger(__name__)

lowerCAmelCase					:     Tuple												={
    '''facebook/vit-mae-base''': '''https://huggingface.co/facebook/vit-mae-base/resolve/main/config.json''',
    # See all ViT MAE models at https://huggingface.co/models?filter=vit-mae
}
class       a_   (				_lowerCAmelCase							):
     __A      =			"vit_mae"





     def __init__( self		:					Any	,      lowercase		:					int=768	,      lowercase		:					Tuple=12	,      lowercase		:					str=12	,      lowercase		:					Optional[Any]=3_072	,      lowercase		:					List[Any]="gelu"	,      lowercase		:					Tuple=0.0	,      lowercase		:					Union[str, Any]=0.0	,      lowercase		:					str=0.02	,      lowercase		:					Optional[int]=1e-1_2	,      lowercase		:					List[Any]=224	,      lowercase		:					str=16	,      lowercase		:					List[str]=3	,      lowercase		:					Optional[Any]=True	,      lowercase		:					int=16	,      lowercase		:					Optional[Any]=512	,      lowercase		:					Optional[Any]=8	,      lowercase		:					Optional[Any]=2_048	,      lowercase		:					List[str]=0.75	,      lowercase		:					str=False	,      **lowercase		:					Union[str, Any]	,      ):





        """simple docstring"""




        super().__init__(**lowercase						)

        lowercase_			:Any    			= hidden_size
        lowercase_			:Optional[Any]    			= num_hidden_layers
        lowercase_			:Optional[Any]    			= num_attention_heads
        lowercase_			:int    			= intermediate_size
        lowercase_			:Optional[int]    			= hidden_act
        lowercase_			:str    			= hidden_dropout_prob
        lowercase_			:Optional[Any]    			= attention_probs_dropout_prob
        lowercase_			:str    			= initializer_range
        lowercase_			:Optional[int]    			= layer_norm_eps
        lowercase_			:str    			= image_size
        lowercase_			:Union[str, Any]    			= patch_size
        lowercase_			:Dict    			= num_channels
        lowercase_			:Any    			= qkv_bias
        lowercase_			:Optional[int]    			= decoder_num_attention_heads
        lowercase_			:Optional[Any]    			= decoder_hidden_size
        lowercase_			:Union[str, Any]    			= decoder_num_hidden_layers
        lowercase_			:List[Any]    			= decoder_intermediate_size
        lowercase_			:Optional[Any]    			= mask_ratio
        lowercase_			:Optional[Any]    			= norm_pix_loss


