


import unittest

from transformers import DebertaConfig, is_torch_available
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
from ...test_pipeline_mixin import PipelineTesterMixin


if is_torch_available():
	import torch

	from transformers import (
	    DebertaForMaskedLM,
	    DebertaForQuestionAnswering,
	    DebertaForSequenceClassification,
	    DebertaForTokenClassification,
	    DebertaModel,
	)
	from transformers.models.deberta.modeling_deberta import DEBERTA_PRETRAINED_MODEL_ARCHIVE_LIST






class      UpperCamelCase(	_a ):


					def __init__(    self						:		str  ,     SCREAMING_SNAKE_CASE						:		Dict  ,     SCREAMING_SNAKE_CASE						:		Optional[Any]=1_3  ,     SCREAMING_SNAKE_CASE						:		List[str]=7  ,     SCREAMING_SNAKE_CASE						:		Any=True  ,     SCREAMING_SNAKE_CASE						:		List[Any]=True  ,     SCREAMING_SNAKE_CASE						:		List[str]=True  ,     SCREAMING_SNAKE_CASE						:		Any=True  ,     SCREAMING_SNAKE_CASE						:		str=9_9  ,     SCREAMING_SNAKE_CASE						:		List[str]=3_2  ,     SCREAMING_SNAKE_CASE						:		int=5  ,     SCREAMING_SNAKE_CASE						:		Union[str, Any]=4  ,     SCREAMING_SNAKE_CASE						:		Union[str, Any]=3_7  ,     SCREAMING_SNAKE_CASE						:		Optional[int]="gelu"  ,     SCREAMING_SNAKE_CASE						:		Dict=0.1  ,     SCREAMING_SNAKE_CASE						:		int=0.1  ,     SCREAMING_SNAKE_CASE						:		int=5_1_2  ,     SCREAMING_SNAKE_CASE						:		int=1_6  ,     SCREAMING_SNAKE_CASE						:		Any=2  ,     SCREAMING_SNAKE_CASE						:		Optional[Any]=0.02  ,     SCREAMING_SNAKE_CASE						:		Tuple=False  ,     SCREAMING_SNAKE_CASE						:		List[Any]=True  ,     SCREAMING_SNAKE_CASE						:		List[Any]="None"  ,     SCREAMING_SNAKE_CASE						:		Dict=3  ,     SCREAMING_SNAKE_CASE						:		Dict=4  ,     SCREAMING_SNAKE_CASE						:		Optional[int]=None  ,     )      ->						List[str]:




											'''simple docstring'''

											__snake_case				=		parent
											__snake_case				=		batch_size
											__snake_case				=		seq_length
											__snake_case				=		is_training
											__snake_case				=		use_input_mask
											__snake_case				=		use_token_type_ids
											__snake_case				=		use_labels
											__snake_case				=		vocab_size
											__snake_case				=		hidden_size
											__snake_case				=		num_hidden_layers
											__snake_case				=		num_attention_heads
											__snake_case				=		intermediate_size
											__snake_case				=		hidden_act
											__snake_case				=		hidden_dropout_prob
											__snake_case				=		attention_probs_dropout_prob
											__snake_case				=		max_position_embeddings
											__snake_case				=		type_vocab_size
											__snake_case				=		type_sequence_label_size
											__snake_case				=		initializer_range
											__snake_case				=		num_labels
											__snake_case				=		num_choices
											__snake_case				=		relative_attention
											__snake_case				=		position_biased_input
											__snake_case				=		pos_att_type
											__snake_case				=		scope


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




											'''simple docstring'''

											__snake_case				=		ids_tensor([self.batch_size, self.seq_length]  ,     self.vocab_size    )

											__snake_case				=		None
											if self.use_input_mask:
																	__snake_case				=		ids_tensor([self.batch_size, self.seq_length]  ,     vocab_size=2    )

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

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

											__snake_case				=		self.get_config()

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


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




											'''simple docstring'''

											return DebertaConfig(
											    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  ,     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  ,     initializer_range=self.initializer_range  ,     relative_attention=self.relative_attention  ,     position_biased_input=self.position_biased_input  ,     pos_att_type=self.pos_att_type  ,     )


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




											'''simple docstring'''

											__snake_case				=		self.get_config()
											__snake_case				=		3_0_0
											return config


					def 				SCREAMING_SNAKE_CASE_	(    self						:		int  ,     SCREAMING_SNAKE_CASE						:		Union[str, Any]    )      ->						Optional[Any]:




											'''simple docstring'''

											self.parent.assertListEqual(list(result.loss.size()    )  ,     []    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Union[str, Any]  ,     SCREAMING_SNAKE_CASE						:		int  ,     SCREAMING_SNAKE_CASE						:		Tuple  ,     SCREAMING_SNAKE_CASE						:		Optional[Any]  ,     SCREAMING_SNAKE_CASE						:		Any  ,     SCREAMING_SNAKE_CASE						:		Union[str, Any]  ,     SCREAMING_SNAKE_CASE						:		Union[str, Any]  ,     SCREAMING_SNAKE_CASE						:		Any    )      ->						Any:




											'''simple docstring'''

											__snake_case				=		DebertaModel(config=SCREAMING_SNAKE_CASE    )
											model.to(SCREAMING_SNAKE_CASE    )
											model.eval()
											__snake_case				=		model(SCREAMING_SNAKE_CASE  ,     attention_mask=SCREAMING_SNAKE_CASE  ,     token_type_ids=SCREAMING_SNAKE_CASE    )[0]
											__snake_case				=		model(SCREAMING_SNAKE_CASE  ,     token_type_ids=SCREAMING_SNAKE_CASE    )[0]
											__snake_case				=		model(SCREAMING_SNAKE_CASE    )[0]

											self.parent.assertListEqual(list(sequence_output.size()    )  ,     [self.batch_size, self.seq_length, self.hidden_size]    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		int  ,     SCREAMING_SNAKE_CASE						:		Dict  ,     SCREAMING_SNAKE_CASE						:		Optional[Any]  ,     SCREAMING_SNAKE_CASE						:		Optional[int]  ,     SCREAMING_SNAKE_CASE						:		Union[str, Any]  ,     SCREAMING_SNAKE_CASE						:		str  ,     SCREAMING_SNAKE_CASE						:		Optional[int]  ,     SCREAMING_SNAKE_CASE						:		Any    )      ->						List[Any]:




											'''simple docstring'''

											__snake_case				=		DebertaForMaskedLM(config=SCREAMING_SNAKE_CASE    )
											model.to(SCREAMING_SNAKE_CASE    )
											model.eval()
											__snake_case				=		model(SCREAMING_SNAKE_CASE  ,     attention_mask=SCREAMING_SNAKE_CASE  ,     token_type_ids=SCREAMING_SNAKE_CASE  ,     labels=SCREAMING_SNAKE_CASE    )

											self.parent.assertEqual(result.logits.shape  ,     (self.batch_size, self.seq_length, self.vocab_size)    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Any  ,     SCREAMING_SNAKE_CASE						:		List[str]  ,     SCREAMING_SNAKE_CASE						:		Optional[int]  ,     SCREAMING_SNAKE_CASE						:		int  ,     SCREAMING_SNAKE_CASE						:		str  ,     SCREAMING_SNAKE_CASE						:		Any  ,     SCREAMING_SNAKE_CASE						:		Tuple  ,     SCREAMING_SNAKE_CASE						:		Union[str, Any]    )      ->						str:




											'''simple docstring'''

											__snake_case				=		self.num_labels
											__snake_case				=		DebertaForSequenceClassification(SCREAMING_SNAKE_CASE    )
											model.to(SCREAMING_SNAKE_CASE    )
											model.eval()
											__snake_case				=		model(SCREAMING_SNAKE_CASE  ,     attention_mask=SCREAMING_SNAKE_CASE  ,     token_type_ids=SCREAMING_SNAKE_CASE  ,     labels=SCREAMING_SNAKE_CASE    )
											self.parent.assertListEqual(list(result.logits.size()    )  ,     [self.batch_size, self.num_labels]    )
											self.check_loss_output(SCREAMING_SNAKE_CASE    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		List[str]  ,     SCREAMING_SNAKE_CASE						:		Any  ,     SCREAMING_SNAKE_CASE						:		Any  ,     SCREAMING_SNAKE_CASE						:		Optional[int]  ,     SCREAMING_SNAKE_CASE						:		int  ,     SCREAMING_SNAKE_CASE						:		Dict  ,     SCREAMING_SNAKE_CASE						:		Any  ,     SCREAMING_SNAKE_CASE						:		Optional[Any]    )      ->						Tuple:




											'''simple docstring'''

											__snake_case				=		self.num_labels
											__snake_case				=		DebertaForTokenClassification(config=SCREAMING_SNAKE_CASE    )
											model.to(SCREAMING_SNAKE_CASE    )
											model.eval()
											__snake_case				=		model(SCREAMING_SNAKE_CASE  ,     attention_mask=SCREAMING_SNAKE_CASE  ,     token_type_ids=SCREAMING_SNAKE_CASE  ,     labels=SCREAMING_SNAKE_CASE    )
											self.parent.assertEqual(result.logits.shape  ,     (self.batch_size, self.seq_length, self.num_labels)    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		str  ,     SCREAMING_SNAKE_CASE						:		Optional[Any]  ,     SCREAMING_SNAKE_CASE						:		Union[str, Any]  ,     SCREAMING_SNAKE_CASE						:		Dict  ,     SCREAMING_SNAKE_CASE						:		List[str]  ,     SCREAMING_SNAKE_CASE						:		int  ,     SCREAMING_SNAKE_CASE						:		int  ,     SCREAMING_SNAKE_CASE						:		Tuple    )      ->						List[str]:




											'''simple docstring'''

											__snake_case				=		DebertaForQuestionAnswering(config=SCREAMING_SNAKE_CASE    )
											model.to(SCREAMING_SNAKE_CASE    )
											model.eval()
											__snake_case				=		model(
											    SCREAMING_SNAKE_CASE  ,     attention_mask=SCREAMING_SNAKE_CASE  ,     token_type_ids=SCREAMING_SNAKE_CASE  ,     start_positions=SCREAMING_SNAKE_CASE  ,     end_positions=SCREAMING_SNAKE_CASE  ,     )
											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 				SCREAMING_SNAKE_CASE_	(    self						:		Optional[int]    )      ->						str:




											'''simple docstring'''

											__snake_case				=		self.prepare_config_and_inputs()
											(
											    (
											    __snake_case
											)		,       (
											    __snake_case
											)		,       (
											    __snake_case
											)		,       (
											    __snake_case
											)		,       (
											    __snake_case
											)		,       (
											    __snake_case
											)		,       (
											    __snake_case
											)		,       
											)				=		config_and_inputs
											__snake_case				=		{"input_ids": input_ids, "token_type_ids": token_type_ids, "attention_mask": input_mask}
											return config, inputs_dict






@require_torch
class      UpperCamelCase(	_a  ,				_a  ,				unittest.TestCase ):
					snake_case_		:					Optional[Any]    					=						(
					    (
					        DebertaModel,
					        DebertaForMaskedLM,
					        DebertaForSequenceClassification,
					        DebertaForTokenClassification,
					        DebertaForQuestionAnswering,
					    )
					    if is_torch_available()
					    else ()
					)
					snake_case_		:					Any    					=						(
					    {
					        """feature-extraction""": DebertaModel,
					        """fill-mask""": DebertaForMaskedLM,
					        """question-answering""": DebertaForQuestionAnswering,
					        """text-classification""": DebertaForSequenceClassification,
					        """token-classification""": DebertaForTokenClassification,
					        """zero-shot""": DebertaForSequenceClassification,
					    }
					    if is_torch_available()
					    else {}
					)

					snake_case_		:					List[str]    					=						True
					snake_case_		:					List[Any]    					=						False
					snake_case_		:					int    					=						False
					snake_case_		:					Dict    					=						False
					snake_case_		:					List[Any]    					=						False


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Tuple    )      ->						Optional[Any]:




											'''simple docstring'''

											__snake_case				=		DebertaModelTester(self    )
											__snake_case				=		ConfigTester(self  ,     config_class=SCREAMING_SNAKE_CASE  ,     hidden_size=3_7    )


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




											'''simple docstring'''

											self.config_tester.run_common_tests()


					def 				SCREAMING_SNAKE_CASE_	(    self						:		int    )      ->						Tuple:




											'''simple docstring'''

											__snake_case				=		self.model_tester.prepare_config_and_inputs()
											self.model_tester.create_and_check_deberta_model(*SCREAMING_SNAKE_CASE    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Any    )      ->						Optional[Any]:




											'''simple docstring'''

											__snake_case				=		self.model_tester.prepare_config_and_inputs()
											self.model_tester.create_and_check_deberta_for_sequence_classification(*SCREAMING_SNAKE_CASE    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Tuple    )      ->						int:




											'''simple docstring'''

											__snake_case				=		self.model_tester.prepare_config_and_inputs()
											self.model_tester.create_and_check_deberta_for_masked_lm(*SCREAMING_SNAKE_CASE    )


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




											'''simple docstring'''

											__snake_case				=		self.model_tester.prepare_config_and_inputs()
											self.model_tester.create_and_check_deberta_for_question_answering(*SCREAMING_SNAKE_CASE    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		str    )      ->						Optional[int]:




											'''simple docstring'''

											__snake_case				=		self.model_tester.prepare_config_and_inputs()
											self.model_tester.create_and_check_deberta_for_token_classification(*SCREAMING_SNAKE_CASE    )






					@slow
					def 				SCREAMING_SNAKE_CASE_	(    self						:		List[str]    )      ->						Optional[int]:




											'''simple docstring'''

											for model_name in DEBERTA_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
																	__snake_case				=		DebertaModel.from_pretrained(SCREAMING_SNAKE_CASE    )
																	self.assertIsNotNone(SCREAMING_SNAKE_CASE    )






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


					@unittest.skip(reason="Model not available yet"    )
					def 				SCREAMING_SNAKE_CASE_	(    self						:		Dict    )      ->						List[str]:




											'''simple docstring'''

											pass






					@slow
					def 				SCREAMING_SNAKE_CASE_	(    self						:		List[str]    )      ->						Dict:




											'''simple docstring'''

											__snake_case				=		DebertaModel.from_pretrained("microsoft/deberta-base"    )

											__snake_case				=		torch.tensor([[0, 3_1_4_1_4, 2_3_2, 3_2_8, 7_4_0, 1_1_4_0, 1_2_6_9_5, 6_9, 4_6_0_7_8, 1_5_8_8, 2]]    )
											__snake_case				=		torch.tensor([[0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]]    )
											with torch.no_grad():
																	__snake_case				=		model(SCREAMING_SNAKE_CASE  ,     attention_mask=SCREAMING_SNAKE_CASE    )[0]
											# compare the actual values for a slice.
											__snake_case				=		torch.tensor(
											    [[[-0.5986, -0.8055, -0.8462], [1.4484, -0.9348, -0.8059], [0.3123, 0.0032, -1.4131]]]    )
											self.assertTrue(torch.allclose(output[:, 1:4, 1:4]  ,     SCREAMING_SNAKE_CASE  ,     atol=1e-4    )  ,     f'''{output[:, 1:4, 1:4]}'''    )









import gc
import unittest

from transformers import MODEL_FOR_MASKED_LM_MAPPING, TF_MODEL_FOR_MASKED_LM_MAPPING, FillMaskPipeline, pipeline
from transformers.pipelines import PipelineException
from transformers.testing_utils import (
    is_pipeline_test,
    is_torch_available,
    nested_simplify,
    require_tf,
    require_torch,
    require_torch_gpu,
    slow,
)

from .test_pipelines_common import ANY






@is_pipeline_test
class      UpperCamelCase(	unittest.TestCase ):
					snake_case_		:					int    					=						MODEL_FOR_MASKED_LM_MAPPING
					snake_case_		:					Optional[int]    					=						TF_MODEL_FOR_MASKED_LM_MAPPING


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




											'''simple docstring'''

											super().tearDown()
											# clean-up as much as possible GPU memory occupied by PyTorch
											gc.collect()
											if is_torch_available():
																	import torch

																	torch.cuda.empty_cache()


					@require_tf
					def 				SCREAMING_SNAKE_CASE_	(    self						:		List[Any]    )      ->						Union[str, Any]:




											'''simple docstring'''

											__snake_case				=		pipeline(task="fill-mask"  ,     model="sshleifer/tiny-distilroberta-base"  ,     top_k=2  ,     framework="tf"    )
											__snake_case				=		unmasker("My name is <mask>"    )
											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE  ,     decimals=6    )  ,     [
											        {"sequence": "My name is grouped", "score": 2.1e-0_5, "token": 3_8_0_1_5, "token_str": " grouped"},
											        {"sequence": "My name is accuser", "score": 2.1e-0_5, "token": 2_5_5_0_6, "token_str": " accuser"},
											    ]  ,     )

											__snake_case				=		unmasker("The largest city in France is <mask>"    )
											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE  ,     decimals=6    )  ,     [
											        {
											            "sequence": "The largest city in France is grouped",
											            "score": 2.1e-0_5,
											            "token": 3_8_0_1_5,
											            "token_str": " grouped",
											        },
											        {
											            "sequence": "The largest city in France is accuser",
											            "score": 2.1e-0_5,
											            "token": 2_5_5_0_6,
											            "token_str": " accuser",
											        },
											    ]  ,     )

											__snake_case				=		unmasker("My name is <mask>"  ,     targets=[" Patrick", " Clara", " Teven"]  ,     top_k=3    )
											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE  ,     decimals=6    )  ,     [
											        {"sequence": "My name is Clara", "score": 2e-0_5, "token": 1_3_6_0_6, "token_str": " Clara"},
											        {"sequence": "My name is Patrick", "score": 2e-0_5, "token": 3_4_9_9, "token_str": " Patrick"},
											        {"sequence": "My name is Te", "score": 1.9e-0_5, "token": 2_9_4_1, "token_str": " Te"},
											    ]  ,     )


					@require_torch
					def 				SCREAMING_SNAKE_CASE_	(    self						:		List[str]    )      ->						List[str]:




											'''simple docstring'''

											__snake_case				=		pipeline(task="fill-mask"  ,     model="sshleifer/tiny-distilroberta-base"  ,     top_k=2  ,     framework="pt"    )

											__snake_case				=		unmasker("My name is <mask>"    )
											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE  ,     decimals=6    )  ,     [
											        {"sequence": "My name is Maul", "score": 2.2e-0_5, "token": 3_5_6_7_6, "token_str": " Maul"},
											        {"sequence": "My name isELS", "score": 2.2e-0_5, "token": 1_6_4_1_6, "token_str": "ELS"},
											    ]  ,     )

											__snake_case				=		unmasker("The largest city in France is <mask>"    )
											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE  ,     decimals=6    )  ,     [
											        {
											            "sequence": "The largest city in France is Maul",
											            "score": 2.2e-0_5,
											            "token": 3_5_6_7_6,
											            "token_str": " Maul",
											        },
											        {"sequence": "The largest city in France isELS", "score": 2.2e-0_5, "token": 1_6_4_1_6, "token_str": "ELS"},
											    ]  ,     )

											__snake_case				=		unmasker("My name is <mask>"  ,     targets=[" Patrick", " Clara", " Teven"]  ,     top_k=3    )
											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE  ,     decimals=6    )  ,     [
											        {"sequence": "My name is Patrick", "score": 2.1e-0_5, "token": 3_4_9_9, "token_str": " Patrick"},
											        {"sequence": "My name is Te", "score": 2e-0_5, "token": 2_9_4_1, "token_str": " Te"},
											        {"sequence": "My name is Clara", "score": 2e-0_5, "token": 1_3_6_0_6, "token_str": " Clara"},
											    ]  ,     )

											__snake_case				=		unmasker("My name is <mask> <mask>"  ,     top_k=2    )

											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE  ,     decimals=6    )  ,     [
											        [
											            {
											                "score": 2.2e-0_5,
											                "token": 3_5_6_7_6,
											                "token_str": " Maul",
											                "sequence": "<s>My name is Maul<mask></s>",
											            },
											            {"score": 2.2e-0_5, "token": 1_6_4_1_6, "token_str": "ELS", "sequence": "<s>My name isELS<mask></s>"},
											        ],
											        [
											            {
											                "score": 2.2e-0_5,
											                "token": 3_5_6_7_6,
											                "token_str": " Maul",
											                "sequence": "<s>My name is<mask> Maul</s>",
											            },
											            {"score": 2.2e-0_5, "token": 1_6_4_1_6, "token_str": "ELS", "sequence": "<s>My name is<mask>ELS</s>"},
											        ],
											    ]  ,     )


					@require_torch_gpu
					def 				SCREAMING_SNAKE_CASE_	(    self						:		Optional[Any]    )      ->						Union[str, Any]:




											'''simple docstring'''

											__snake_case				=		pipeline("fill-mask"  ,     model="hf-internal-testing/tiny-random-distilbert"  ,     device=0  ,     framework="pt"    )

											# convert model to fp16
											pipe.model.half()

											__snake_case				=		pipe("Paris is the [MASK] of France."    )
											# We actually don't care about the result, we just want to make sure
											# it works, meaning the float16 tensor got casted back to float32
											# for postprocessing.
											self.assertIsInstance(SCREAMING_SNAKE_CASE  ,     SCREAMING_SNAKE_CASE    )


					@slow
					@require_torch
					def 				SCREAMING_SNAKE_CASE_	(    self						:		List[str]    )      ->						Tuple:




											'''simple docstring'''

											__snake_case				=		pipeline(task="fill-mask"  ,     model="distilroberta-base"  ,     top_k=2  ,     framework="pt"    )
											self.run_large_test(SCREAMING_SNAKE_CASE    )


					@slow
					@require_tf
					def 				SCREAMING_SNAKE_CASE_	(    self						:		Optional[Any]    )      ->						int:




											'''simple docstring'''

											__snake_case				=		pipeline(task="fill-mask"  ,     model="distilroberta-base"  ,     top_k=2  ,     framework="tf"    )
											self.run_large_test(SCREAMING_SNAKE_CASE    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Any  ,     SCREAMING_SNAKE_CASE						:		Optional[Any]    )      ->						str:




											'''simple docstring'''

											__snake_case				=		unmasker("My name is <mask>"    )
											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE    )  ,     [
											        {"sequence": "My name is John", "score": 0.008, "token": 6_1_0, "token_str": " John"},
											        {"sequence": "My name is Chris", "score": 0.007, "token": 1_5_7_3, "token_str": " Chris"},
											    ]  ,     )
											__snake_case				=		unmasker("The largest city in France is <mask>"    )
											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE    )  ,     [
											        {
											            "sequence": "The largest city in France is Paris",
											            "score": 0.251,
											            "token": 2_2_0_1,
											            "token_str": " Paris",
											        },
											        {
											            "sequence": "The largest city in France is Lyon",
											            "score": 0.214,
											            "token": 1_2_7_9_0,
											            "token_str": " Lyon",
											        },
											    ]  ,     )

											__snake_case				=		unmasker("My name is <mask>"  ,     targets=[" Patrick", " Clara", " Teven"]  ,     top_k=3    )
											self.assertEqual(
											    nested_simplify(SCREAMING_SNAKE_CASE    )  ,     [
											        {"sequence": "My name is Patrick", "score": 0.005, "token": 3_4_9_9, "token_str": " Patrick"},
											        {"sequence": "My name is Clara", "score": 0.000, "token": 1_3_6_0_6, "token_str": " Clara"},
											        {"sequence": "My name is Te", "score": 0.000, "token": 2_9_4_1, "token_str": " Te"},
											    ]  ,     )


					@require_torch
					def 				SCREAMING_SNAKE_CASE_	(    self						:		Dict    )      ->						str:




											'''simple docstring'''

											__snake_case				=		pipeline(task="fill-mask"  ,     model="sshleifer/tiny-distilroberta-base"  ,     framework="pt"    )
											__snake_case				=		None
											__snake_case				=		None
											self.run_pipeline_test(SCREAMING_SNAKE_CASE  ,     []    )


					@require_tf
					def 				SCREAMING_SNAKE_CASE_	(    self						:		List[Any]    )      ->						Optional[int]:




											'''simple docstring'''

											__snake_case				=		pipeline(task="fill-mask"  ,     model="sshleifer/tiny-distilroberta-base"  ,     framework="tf"    )
											__snake_case				=		None
											__snake_case				=		None
											self.run_pipeline_test(SCREAMING_SNAKE_CASE  ,     []    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Tuple  ,     SCREAMING_SNAKE_CASE						:		Any  ,     SCREAMING_SNAKE_CASE						:		Optional[Any]  ,     SCREAMING_SNAKE_CASE						:		Tuple    )      ->						int:




											'''simple docstring'''

											if tokenizer is None or tokenizer.mask_token_id is None:
																	self.skipTest("The provided tokenizer has no mask token, (probably reformer or wav2vec2)"    )

											__snake_case				=		FillMaskPipeline(model=SCREAMING_SNAKE_CASE  ,     tokenizer=SCREAMING_SNAKE_CASE    )
											__snake_case				=		[
											    f'''This is another {tokenizer.mask_token} test''',
											]
											return fill_masker, examples


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Any  ,     SCREAMING_SNAKE_CASE						:		int  ,     SCREAMING_SNAKE_CASE						:		int    )      ->						Optional[Any]:




											'''simple docstring'''

											__snake_case				=		fill_masker.tokenizer
											__snake_case				=		fill_masker.model

											__snake_case				=		fill_masker(
											    f'''This is a {tokenizer.mask_token}'''  ,     )
											self.assertEqual(
											    SCREAMING_SNAKE_CASE  ,     [
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											    ]  ,     )

											__snake_case				=		fill_masker([f'''This is a {tokenizer.mask_token}''']    )
											self.assertEqual(
											    SCREAMING_SNAKE_CASE  ,     [
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											    ]  ,     )

											__snake_case				=		fill_masker([f'''This is a {tokenizer.mask_token}''', f'''Another {tokenizer.mask_token} great test.''']    )
											self.assertEqual(
											    SCREAMING_SNAKE_CASE  ,     [
											        [
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        ],
											        [
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        ],
											    ]  ,     )

											with self.assertRaises(SCREAMING_SNAKE_CASE    ):
																	fill_masker([None]    )
											# No mask_token is not supported
											with self.assertRaises(SCREAMING_SNAKE_CASE    ):
																	fill_masker("This is"    )

											self.run_test_top_k(SCREAMING_SNAKE_CASE  ,     SCREAMING_SNAKE_CASE    )
											self.run_test_targets(SCREAMING_SNAKE_CASE  ,     SCREAMING_SNAKE_CASE    )
											self.run_test_top_k_targets(SCREAMING_SNAKE_CASE  ,     SCREAMING_SNAKE_CASE    )
											self.fill_mask_with_duplicate_targets_and_top_k(SCREAMING_SNAKE_CASE  ,     SCREAMING_SNAKE_CASE    )
											self.fill_mask_with_multiple_masks(SCREAMING_SNAKE_CASE  ,     SCREAMING_SNAKE_CASE    )


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




											'''simple docstring'''

											__snake_case				=		tokenizer.get_vocab()
											__snake_case				=		sorted(vocab.keys()    )[:2]
											# Pipeline argument
											__snake_case				=		FillMaskPipeline(model=SCREAMING_SNAKE_CASE  ,     tokenizer=SCREAMING_SNAKE_CASE  ,     targets=SCREAMING_SNAKE_CASE    )
											__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''    )
											self.assertEqual(
											    SCREAMING_SNAKE_CASE  ,     [
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											    ]  ,     )
											__snake_case				=		{vocab[el] for el in targets}
											self.assertEqual({el["token"] for el in outputs}  ,     SCREAMING_SNAKE_CASE    )
											__snake_case				=		[tokenizer.decode([x]    ) for x in target_ids]
											self.assertEqual({el["token_str"] for el in outputs}  ,     set(SCREAMING_SNAKE_CASE    )    )

											# Call argument
											__snake_case				=		FillMaskPipeline(model=SCREAMING_SNAKE_CASE  ,     tokenizer=SCREAMING_SNAKE_CASE    )
											__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''  ,     targets=SCREAMING_SNAKE_CASE    )
											self.assertEqual(
											    SCREAMING_SNAKE_CASE  ,     [
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											    ]  ,     )
											__snake_case				=		{vocab[el] for el in targets}
											self.assertEqual({el["token"] for el in outputs}  ,     SCREAMING_SNAKE_CASE    )
											__snake_case				=		[tokenizer.decode([x]    ) for x in target_ids]
											self.assertEqual({el["token_str"] for el in outputs}  ,     set(SCREAMING_SNAKE_CASE    )    )

											# Score equivalence
											__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''  ,     targets=SCREAMING_SNAKE_CASE    )
											__snake_case				=		[top_mask["token_str"] for top_mask in outputs]
											__snake_case				=		[top_mask["score"] for top_mask in outputs]

											# For some BPE tokenizers, `</w>` is removed during decoding, so `token_str` won't be the same as in `targets`.
											if set(SCREAMING_SNAKE_CASE    ) == set(SCREAMING_SNAKE_CASE    ):
																	__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''  ,     targets=SCREAMING_SNAKE_CASE    )
																	__snake_case				=		[top_mask["score"] for top_mask in unmasked_targets]
																	self.assertEqual(nested_simplify(SCREAMING_SNAKE_CASE    )  ,     nested_simplify(SCREAMING_SNAKE_CASE    )    )

											# Raises with invalid
											with self.assertRaises(SCREAMING_SNAKE_CASE    ):
																	__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''  ,     targets=[]    )
											# For some tokenizers, `""` is actually in the vocabulary and the expected error won't raised
											if "" not in tokenizer.get_vocab():
																	with self.assertRaises(SCREAMING_SNAKE_CASE    ):
																							__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''  ,     targets=[""]    )
																	with self.assertRaises(SCREAMING_SNAKE_CASE    ):
																							__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''  ,     targets=""    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Optional[int]  ,     SCREAMING_SNAKE_CASE						:		Union[str, Any]  ,     SCREAMING_SNAKE_CASE						:		List[str]    )      ->						Optional[Any]:




											'''simple docstring'''

											__snake_case				=		FillMaskPipeline(model=SCREAMING_SNAKE_CASE  ,     tokenizer=SCREAMING_SNAKE_CASE  ,     top_k=2    )
											__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''    )
											self.assertEqual(
											    SCREAMING_SNAKE_CASE  ,     [
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											    ]  ,     )

											__snake_case				=		FillMaskPipeline(model=SCREAMING_SNAKE_CASE  ,     tokenizer=SCREAMING_SNAKE_CASE    )
											__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''  ,     top_k=2    )
											self.assertEqual(
											    SCREAMING_SNAKE_CASE  ,     [
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											    ]  ,     )
											self.assertEqual(nested_simplify(SCREAMING_SNAKE_CASE    )  ,     nested_simplify(SCREAMING_SNAKE_CASE    )    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		List[Any]  ,     SCREAMING_SNAKE_CASE						:		str  ,     SCREAMING_SNAKE_CASE						:		List[Any]    )      ->						Dict:




											'''simple docstring'''

											__snake_case				=		tokenizer.get_vocab()
											__snake_case				=		FillMaskPipeline(model=SCREAMING_SNAKE_CASE  ,     tokenizer=SCREAMING_SNAKE_CASE    )

											# top_k=2, ntargets=3
											__snake_case				=		sorted(vocab.keys()    )[:3]
											__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''  ,     top_k=2  ,     targets=SCREAMING_SNAKE_CASE    )

											# If we use the most probably targets, and filter differently, we should still
											# have the same results
											__snake_case				=		[el["token_str"] for el in sorted(SCREAMING_SNAKE_CASE  ,     key=lambda SCREAMING_SNAKE_CASE    : x["score"]  ,     reverse=SCREAMING_SNAKE_CASE    )]
											# For some BPE tokenizers, `</w>` is removed during decoding, so `token_str` won't be the same as in `targets`.
											if set(SCREAMING_SNAKE_CASE    ).issubset(SCREAMING_SNAKE_CASE    ):
																	__snake_case				=		fill_masker(f'''This is a {tokenizer.mask_token}'''  ,     top_k=3  ,     targets=SCREAMING_SNAKE_CASE    )
																	# They should yield exactly the same result
																	self.assertEqual(nested_simplify(SCREAMING_SNAKE_CASE    )  ,     nested_simplify(SCREAMING_SNAKE_CASE    )    )


					def 				SCREAMING_SNAKE_CASE_	(    self						:		Optional[int]  ,     SCREAMING_SNAKE_CASE						:		Dict  ,     SCREAMING_SNAKE_CASE						:		Dict    )      ->						Optional[Any]:




											'''simple docstring'''

											__snake_case				=		FillMaskPipeline(model=SCREAMING_SNAKE_CASE  ,     tokenizer=SCREAMING_SNAKE_CASE    )
											__snake_case				=		tokenizer.get_vocab()
											# String duplicates + id duplicates
											__snake_case				=		sorted(vocab.keys()    )[:3]
											__snake_case				=		[targets[0], targets[1], targets[0], targets[2], targets[1]]
											__snake_case				=		fill_masker(f'''My name is {tokenizer.mask_token}'''  ,     targets=SCREAMING_SNAKE_CASE  ,     top_k=1_0    )

											# The target list contains duplicates, so we can't output more
											# than them
											self.assertEqual(len(SCREAMING_SNAKE_CASE    )  ,     3    )






					def 				SCREAMING_SNAKE_CASE_	(    self						:		str  ,     SCREAMING_SNAKE_CASE						:		Optional[Any]  ,     SCREAMING_SNAKE_CASE						:		str    )      ->						Optional[Any]:




											'''simple docstring'''

											__snake_case				=		FillMaskPipeline(model=SCREAMING_SNAKE_CASE  ,     tokenizer=SCREAMING_SNAKE_CASE    )

											__snake_case				=		fill_masker(
											    f'''This is a {tokenizer.mask_token} {tokenizer.mask_token} {tokenizer.mask_token}'''  ,     top_k=2    )
											self.assertEqual(
											    SCREAMING_SNAKE_CASE  ,     [
											        [
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        ],
											        [
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        ],
											        [
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											            {"sequence": ANY(SCREAMING_SNAKE_CASE    ), "score": ANY(SCREAMING_SNAKE_CASE    ), "token": ANY(SCREAMING_SNAKE_CASE    ), "token_str": ANY(SCREAMING_SNAKE_CASE    )},
											        ],
											    ]  ,     )





