



'''simple docstring'''






import datasets


UpperCamelCase__:   Tuple								=			"\\n@InProceedings{conneau2018xnli,\n  author = \"Conneau, Alexis\n                 and Rinott, Ruty\n                 and Lample, Guillaume\n                 and Williams, Adina\n                 and Bowman, Samuel R.\n                 and Schwenk, Holger\n                 and Stoyanov, Veselin\",\n  title = \"XNLI: Evaluating Cross-lingual Sentence Representations\",\n  booktitle = \"Proceedings of the 2018 Conference on Empirical Methods\n               in Natural Language Processing\",\n  year = \"2018\",\n  publisher = \"Association for Computational Linguistics\",\n  location = \"Brussels, Belgium\",\n}\n"

UpperCamelCase__:   List[str]								=			"\\nXNLI is a subset of a few thousand examples from MNLI which has been translated\ninto a 14 different languages (some low-ish resource). As with MNLI, the goal is\nto predict textual entailment (does sentence A imply/contradict/neither sentence\nB) and is a classification task (given two sentences, predict one of three\nlabels).\n"

UpperCamelCase__:   List[Any]								=			"\nComputes XNLI score which is just simple accuracy.\nArgs:\n    predictions: Predicted labels.\n    references: Ground truth labels.\nReturns:\n    'accuracy': accuracy\nExamples:\n\n    >>> predictions = [0, 1]\n    >>> references = [0, 1]\n    >>> xnli_metric = datasets.load_metric(\"xnli\")\n    >>> results = xnli_metric.compute(predictions=predictions, references=references)\n    >>> print(results)\n    {'accuracy': 1.0}\n"


def  snake_case_     (					_lowerCAmelCase  :						Tuple   ,      _lowerCAmelCase  :						Tuple  )      ->   List[str]:
							return (preds == labels).mean()




@datasets.utils.file_utils.add_start_docstrings(_DESCRIPTION							,		_KWARGS_DESCRIPTION			)
class 				SCREAMING_SNAKE_CASE(							datasets.Metric			):







	"""simple docstring"""






	def      A	(						self			:     List[Any]		)			->		Optional[int]:
								return datasets.MetricInfo(
								    description=_DESCRIPTION		,			citation=_CITATION		,			inputs_description=_KWARGS_DESCRIPTION		,			features=datasets.Features(
								        {
								            '''predictions''': datasets.Value('''int64''' if self.config_name != '''sts-b''' else '''float32'''		),
								            '''references''': datasets.Value('''int64''' if self.config_name != '''sts-b''' else '''float32'''		),
								        }		)		,			codebase_urls=[]		,			reference_urls=[]		,			format='''numpy'''		,			)







	def      A	(						self			:     Dict		,			__snake_case			:     Union[str, Any]		,			__snake_case			:     Union[str, Any]		)			->		List[str]:
								return {"accuracy": simple_accuracy(__snake_case		,			__snake_case		)}







import random
import unittest

import torch

from diffusers import IFInpaintingPipeline
from diffusers.utils import floats_tensor
from diffusers.utils.import_utils import is_xformers_available
from diffusers.utils.testing_utils import skip_mps, torch_device

from ..pipeline_params import (
    TEXT_GUIDED_IMAGE_INPAINTING_BATCH_PARAMS,
    TEXT_GUIDED_IMAGE_INPAINTING_PARAMS,
)
from ..test_pipelines_common import PipelineTesterMixin
from . import IFPipelineTesterMixin







@skip_mps
class 				lowercase		(					SCREAMING_SNAKE_CASE__       ,  SCREAMING_SNAKE_CASE__       ,  unittest.TestCase		):
							lowercase_       :       List[Any]       					=IFInpaintingPipeline
							lowercase_       :       Optional[int]       					=TEXT_GUIDED_IMAGE_INPAINTING_PARAMS - {'''width''', '''height'''}
							lowercase_       :       Any       					=TEXT_GUIDED_IMAGE_INPAINTING_BATCH_PARAMS
							lowercase_       :       str       					=PipelineTesterMixin.required_optional_params - {'''latents'''}




							def  A__      (				self):
												return self._get_dummy_components()




							def  A__      (				self   ,A__   ,A__=0):
												if str(A__).startswith('''mps'''):
																	lowercase							=						torch.manual_seed(A__)
												else:
																	lowercase							=						torch.Generator(device=A__).manual_seed(A__)

												lowercase							=						floats_tensor((1, 3, 3_2, 3_2)   ,rng=random.Random(A__)).to(A__)
												lowercase							=						floats_tensor((1, 3, 3_2, 3_2)   ,rng=random.Random(A__)).to(A__)

												lowercase							=						{
												    '''prompt''': '''A painting of a squirrel eating a burger''',
												    '''image''': image,
												    '''mask_image''': mask_image,
												    '''generator''': generator,
												    '''num_inference_steps''': 2,
												    '''output_type''': '''numpy''',
												}

												return inputs




							@unittest.skipIf(
							    torch_device != '''cuda''' or not is_xformers_available()   ,reason='''XFormers attention is only available with CUDA and `xformers` installed'''   ,)
							def  A__      (				self):
												self._test_xformers_attention_forwardGenerator_pass(expected_max_diff=1E-3)




							def  A__      (				self):
												self._test_save_load_optional_components()




							@unittest.skipIf(torch_device != '''cuda'''   ,reason='''float16 requires CUDA''')
							def  A__      (				self):
												# Due to non-determinism in save load of the hf-internal-testing/tiny-random-t5 text encoder
												super().test_save_load_floataa(expected_max_diff=1E-1)




							def  A__      (				self):
												self._test_attention_slicing_forward_pass(expected_max_diff=1E-2)




							def  A__      (				self):
												self._test_save_load_local()





							def  A__      (				self):
												self._test_inference_batch_single_identical(
												    expected_max_diff=1E-2   ,)




