
import numpy as np

import datasets


_A   =     '\nCompute the Mahalanobis Distance\n\nMahalonobis distance is the distance between a point and a distribution.\nAnd not between two distinct points. It is effectively a multivariate equivalent of the Euclidean distance.\nIt was introduced by Prof. P. C. Mahalanobis in 1936\nand has been used in various statistical applications ever since\n[source: https://www.machinelearningplus.com/statistics/mahalanobis-distance/]\n'

_A   =     '\\n@article{de2000mahalanobis,\n  title={The mahalanobis distance},\n  author={De Maesschalck, Roy and Jouan-Rimbaud, Delphine and Massart, D{\'e}sir{\'e} L},\n  journal={Chemometrics and intelligent laboratory systems},\n  volume={50},\n  number={1},\n  pages={1--18},\n  year={2000},\n  publisher={Elsevier}\n}\n'

_A   =     '\nArgs:\n    X: List of datapoints to be compared with the `reference_distribution`.\n    reference_distribution: List of datapoints from the reference distribution we want to compare to.\nReturns:\n    mahalanobis: The Mahalonobis distance for each datapoint in `X`.\nExamples:\n\n    >>> mahalanobis_metric = datasets.load_metric("mahalanobis")\n    >>> results = mahalanobis_metric.compute(reference_distribution=[[0, 1], [1, 0]], X=[[0, 1]])\n    >>> print(results)\n    {\'mahalanobis\': array([0.5])}\n'





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




				def      __a						(					self     )				->      Any:
							return datasets.MetricInfo(
							    description=_DESCRIPTION				,				citation=_CITATION				,				inputs_description=_KWARGS_DESCRIPTION				,				features=datasets.Features(
							        {
							            "X": datasets.Sequence(datasets.Value("float"				,				id="sequence"     )				,				id="X"     ),
							        }     )				,				)




				def      __a						(					self				,				_UpperCamelCase				,				_UpperCamelCase     )				->      Dict:
							lowerCAmelCase_						=	np.array(_SCREAMING_SNAKE_CASE     )
							lowerCAmelCase_						=	np.array(_SCREAMING_SNAKE_CASE     )

							# Assert that arrays are 2D
							if len(X.shape     ) != 2:
										raise ValueError("Expected `X` to be a 2D vector"     )
							if len(reference_distribution.shape     ) != 2:
										raise ValueError("Expected `reference_distribution` to be a 2D vector"     )
							if reference_distribution.shape[0] < 2:
										raise ValueError(
										    "Expected `reference_distribution` to be a 2D vector with more than one element in the first dimension"     )

							# Get mahalanobis distance for each prediction
							lowerCAmelCase_						=	X - np.mean(_SCREAMING_SNAKE_CASE     )
							lowerCAmelCase_						=	np.cov(reference_distribution.T     )
							try:
										lowerCAmelCase_						=	np.linalg.inv(_SCREAMING_SNAKE_CASE     )
							except np.linalg.LinAlgError:
										lowerCAmelCase_						=	np.linalg.pinv(_SCREAMING_SNAKE_CASE     )
							lowerCAmelCase_						=	np.dot(_SCREAMING_SNAKE_CASE				,				_SCREAMING_SNAKE_CASE     )
							lowerCAmelCase_						=	np.dot(_SCREAMING_SNAKE_CASE				,				X_minus_mu.T     ).diagonal()

							return {"mahalanobis": mahal_dist}








"""simple docstring"""






import os
import unittest

from transformers import FunnelTokenizer, FunnelTokenizerFast
from transformers.models.funnel.tokenization_funnel import VOCAB_FILES_NAMES
from transformers.testing_utils import require_tokenizers

from ...test_tokenization_common import TokenizerTesterMixin



@require_tokenizers
class 			A__   (  __SCREAMING_SNAKE_CASE  ,		unittest.TestCase							):




	'''simple docstring'''





	SCREAMING_SNAKE_CASE          =	FunnelTokenizer
	SCREAMING_SNAKE_CASE          =	FunnelTokenizerFast
	SCREAMING_SNAKE_CASE          =	True
	SCREAMING_SNAKE_CASE          =	True

	def 							_SCREAMING_SNAKE_CASE   (     self:		Tuple)				->       Optional[int]:

				"""simple docstring"""

				super().setUp()

				__lowerCAmelCase      :  str												=       [
				    "<unk>",
				    "<cls>",
				    "<sep>",
				    "want",
				    "##want",
				    "##ed",
				    "wa",
				    "un",
				    "runn",
				    "##ing",
				    ",",
				    "low",
				    "lowest",
				]
				__lowerCAmelCase      :  int												=       os.path.join(self.tmpdirname       ,			VOCAB_FILES_NAMES["vocab_file"])
				with open(self.vocab_file       ,			"w"       ,			encoding="utf-8") as vocab_writer:
							vocab_writer.write("".join([x + "\n" for x in vocab_tokens]))

	def 							_SCREAMING_SNAKE_CASE   (     self:		Optional[Any]       ,			**_SCREAMING_SNAKE_CASE:		Union[str, Any])				->       int:

				"""simple docstring"""

				return FunnelTokenizer.from_pretrained(self.tmpdirname       ,			**_SCREAMING_SNAKE_CASE)

	def 							_SCREAMING_SNAKE_CASE   (     self:		Any       ,			**_SCREAMING_SNAKE_CASE:		Any)				->       str:

				"""simple docstring"""

				return FunnelTokenizerFast.from_pretrained(self.tmpdirname       ,			**_SCREAMING_SNAKE_CASE)

	def 							_SCREAMING_SNAKE_CASE   (     self:		Tuple       ,			_SCREAMING_SNAKE_CASE:		str)				->       Any:

				"""simple docstring"""

				__lowerCAmelCase      :  Union[str, Any]												=       "UNwant\u00E9d,running"
				__lowerCAmelCase      :  str												=       "unwanted, running"
				return input_text, output_text

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

				"""simple docstring"""

				__lowerCAmelCase      :  Any												=       self.tokenizer_class(self.vocab_file)

				__lowerCAmelCase      :  Any												=       tokenizer.tokenize("UNwant\u00E9d,running")
				self.assertListEqual(_SCREAMING_SNAKE_CASE       ,			["un", "##want", "##ed", ",", "runn", "##ing"])
				self.assertListEqual(tokenizer.convert_tokens_to_ids(_SCREAMING_SNAKE_CASE)       ,			[7, 4, 5, 10, 8, 9])







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

				"""simple docstring"""

				__lowerCAmelCase      :  Optional[Any]												=       self.get_tokenizers(do_lower_case=_SCREAMING_SNAKE_CASE)
				for tokenizer in tokenizers:
							__lowerCAmelCase      :  List[str]												=       tokenizer("UNwant\u00E9d,running")
							__lowerCAmelCase      :  Optional[int]												=       len(inputs["input_ids"]) - 1
							self.assertListEqual(inputs["token_type_ids"]       ,			[2] + [0] * sentence_len)

							__lowerCAmelCase      :  List[str]												=       tokenizer("UNwant\u00E9d,running"       ,			"UNwant\u00E9d,running")
							self.assertListEqual(inputs["token_type_ids"]       ,			[2] + [0] * sentence_len + [1] * sentence_len)