






import faiss  # noqa: F401 # Here to have a nice missing dependency error message early on
import numpy  # noqa: F401 # Here to have a nice missing dependency error message early on
import requests  # noqa: F401 # Here to have a nice missing dependency error message early on
import sklearn  # noqa: F401 # Here to have a nice missing dependency error message early on
import tqdm  # noqa: F401 # Here to have a nice missing dependency error message early on
from mauve import compute_mauve  # From: mauve-text

import datasets


__magic_name__:					Optional[Any]							=     "\\n@inproceedings{pillutla-etal:mauve:neurips2021,\n  title={MAUVE: Measuring the Gap Between Neural Text and Human Text using Divergence Frontiers},\n  author={Pillutla, Krishna and Swayamdipta, Swabha and Zellers, Rowan and Thickstun, John and Welleck, Sean and Choi, Yejin and Harchaoui, Zaid},\n  booktitle = {NeurIPS},\n  year      = {2021}\n}\n\n"

__magic_name__:					str							=     "\\nMAUVE is a library built on PyTorch and HuggingFace Transformers to measure the gap between neural text and human text with the eponymous MAUVE measure.\n\nMAUVE summarizes both Type I and Type II errors measured softly using Kullback–Leibler (KL) divergences.\n\nFor details, see the MAUVE paper: https://arxiv.org/abs/2102.01454 (Neurips, 2021).\n\nThis metrics is a wrapper around the official implementation of MAUVE:\nhttps://github.com/krishnap25/mauve\n"

__magic_name__:					str							=     "\nCalculates MAUVE scores between two lists of generated text and reference text.\nArgs:\n    predictions: list of generated text to score. Each predictions\n        should be a string with tokens separated by spaces.\n    references: list of reference for each prediction. Each\n        reference should be a string with tokens separated by spaces.\nOptional Args:\n    num_buckets: the size of the histogram to quantize P and Q. Options: 'auto' (default) or an integer\n    pca_max_data: the number data points to use for PCA dimensionality reduction prior to clustering. If -1, use all the data. Default -1\n    kmeans_explained_var: amount of variance of the data to keep in dimensionality reduction by PCA. Default 0.9\n    kmeans_num_redo: number of times to redo k-means clustering (the best objective is kept). Default 5\n    kmeans_max_iter: maximum number of k-means iterations. Default 500\n    featurize_model_name: name of the model from which features are obtained. Default 'gpt2-large' Use one of ['gpt2', 'gpt2-medium', 'gpt2-large', 'gpt2-xl'].\n    device_id: Device for featurization. Supply a GPU id (e.g. 0 or 3) to use GPU. If no GPU with this id is found, use CPU\n    max_text_length: maximum number of tokens to consider. Default 1024\n    divergence_curve_discretization_size: Number of points to consider on the divergence curve. Default 25\n    mauve_scaling_factor: \"c\" from the paper. Default 5.\n    verbose: If True (default), print running time updates\n    seed: random seed to initialize k-means cluster assignments.\nReturns:\n    mauve: MAUVE score, a number between 0 and 1. Larger values indicate that P and Q are closer,\n    frontier_integral: Frontier Integral, a number between 0 and 1. Smaller values indicate that P and Q are closer,\n    divergence_curve: a numpy.ndarray of shape (m, 2); plot it with matplotlib to view the divergence curve,\n    p_hist: a discrete distribution, which is a quantized version of the text distribution p_text,\n    q_hist: same as above, but with q_text.\nExamples:\n\n    >>> # faiss segfaults in doctest for some reason, so the .compute call is not tested with doctest\n    >>> import datasets\n    >>> mauve = datasets.load_metric('mauve')\n    >>> predictions = [\"hello there\", \"general kenobi\"]\n    >>> references = [\"hello there\", \"general kenobi\"]\n    >>> out = mauve.compute(predictions=predictions, references=references) # doctest: +SKIP\n    >>> print(out.mauve) # doctest: +SKIP\n    1.0\n"





@datasets.utils.file_utils.add_start_docstrings(_DESCRIPTION						,  _KWARGS_DESCRIPTION				)
class 	snake_case__ (							datasets.Metric				):
	def       __magic_name__    (    self			)							->				Union[str, Any]:
							return datasets.MetricInfo(
							    description=_DESCRIPTION      ,       citation=_CITATION      ,       homepage="""https://github.com/krishnap25/mauve"""      ,       inputs_description=_KWARGS_DESCRIPTION      ,       features=datasets.Features(
							        {
							            """predictions""": datasets.Value("""string"""      ,       id="""sequence"""			),
							            """references""": datasets.Value("""string"""      ,       id="""sequence"""			),
							        }			)      ,       codebase_urls=["""https://github.com/krishnap25/mauve"""]      ,       reference_urls=[
							        """https://arxiv.org/abs/2102.01454""",
							        """https://github.com/krishnap25/mauve""",
							    ]      ,       )







	def       __magic_name__    (    self      ,       lowerCAmelCase__      ,       lowerCAmelCase__      ,       lowerCAmelCase__=None      ,       lowerCAmelCase__=None      ,       lowerCAmelCase__=None      ,       lowerCAmelCase__=None      ,       lowerCAmelCase__="auto"      ,       lowerCAmelCase__=-1      ,       lowerCAmelCase__=0.9      ,       lowerCAmelCase__=5      ,       lowerCAmelCase__=5_00      ,       lowerCAmelCase__="gpt2-large"      ,       lowerCAmelCase__=-1      ,       lowerCAmelCase__=10_24      ,       lowerCAmelCase__=25      ,       lowerCAmelCase__=5      ,       lowerCAmelCase__=True      ,       lowerCAmelCase__=25      ,       )							->				Union[str, Any]:
							__magic_name__      :       Tuple       						= compute_mauve(
							    p_text=lowerCAmelCase__      ,       q_text=lowerCAmelCase__      ,       p_features=lowerCAmelCase__      ,       q_features=lowerCAmelCase__      ,       p_tokens=lowerCAmelCase__      ,       q_tokens=lowerCAmelCase__      ,       num_buckets=lowerCAmelCase__      ,       pca_max_data=lowerCAmelCase__      ,       kmeans_explained_var=lowerCAmelCase__      ,       kmeans_num_redo=lowerCAmelCase__      ,       kmeans_max_iter=lowerCAmelCase__      ,       featurize_model_name=lowerCAmelCase__      ,       device_id=lowerCAmelCase__      ,       max_text_length=lowerCAmelCase__      ,       divergence_curve_discretization_size=lowerCAmelCase__      ,       mauve_scaling_factor=lowerCAmelCase__      ,       verbose=lowerCAmelCase__      ,       seed=lowerCAmelCase__      ,       )
							return out








import json
import os
import shutil
import tempfile
import unittest

import numpy as np
import pytest

from transformers import CLIPTokenizer, CLIPTokenizerFast
from transformers.models.clip.tokenization_clip import VOCAB_FILES_NAMES
from transformers.testing_utils import require_vision
from transformers.utils import IMAGE_PROCESSOR_NAME, is_vision_available


if is_vision_available():
		from PIL import Image

		from transformers import OwlViTImageProcessor, OwlViTProcessor





@require_vision
class 	snake_case__ (							unittest.TestCase				):
	def       __magic_name__    (    self			)							->				int:
							__magic_name__      :       List[str]       						= tempfile.mkdtemp()

							# fmt: off
							__magic_name__      :       Union[str, Any]       						= ["""""", """l""", """o""", """w""", """e""", """r""", """s""", """t""", """i""", """d""", """n""", """lo""", """l</w>""", """w</w>""", """r</w>""", """t</w>""", """low</w>""", """er</w>""", """lowest</w>""", """newer</w>""", """wider""", """<unk>""", """<|startoftext|>""", """<|endoftext|>"""]
							# fmt: on
							__magic_name__      :       Any       						= dict(zip(lowerCAmelCase__      ,       range(len(lowerCAmelCase__			)			)			)			)
							__magic_name__      :       int       						= ["""#version: 0.2""", """l o""", """lo w</w>""", """e r</w>""", """"""]
							__magic_name__      :       Any       						= {"""unk_token""": """<unk>"""}

							__magic_name__      :       Optional[Any]       						= os.path.join(self.tmpdirname      ,       VOCAB_FILES_NAMES["""vocab_file"""]			)
							__magic_name__      :       Optional[int]       						= os.path.join(self.tmpdirname      ,       VOCAB_FILES_NAMES["""merges_file"""]			)
							with open(self.vocab_file      ,       """w"""      ,       encoding="""utf-8"""			) as fp:
													fp.write(json.dumps(lowerCAmelCase__			) + """\n"""			)
							with open(self.merges_file      ,       """w"""      ,       encoding="""utf-8"""			) as fp:
													fp.write("""\n""".join(lowerCAmelCase__			)			)

							__magic_name__      :       int       						= {
							    """do_resize""": True,
							    """size""": 20,
							    """do_center_crop""": True,
							    """crop_size""": 18,
							    """do_normalize""": True,
							    """image_mean""": [0.4_8_1_4_5_4_6_6, 0.4_5_7_8_2_7_5, 0.4_0_8_2_1_0_7_3],
							    """image_std""": [0.2_6_8_6_2_9_5_4, 0.2_6_1_3_0_2_5_8, 0.2_7_5_7_7_7_1_1],
							}
							__magic_name__      :       List[str]       						= os.path.join(self.tmpdirname      ,       lowerCAmelCase__			)
							with open(self.image_processor_file      ,       """w"""      ,       encoding="""utf-8"""			) as fp:
													json.dump(lowerCAmelCase__      ,       lowerCAmelCase__			)
	def       __magic_name__    (    self      ,       **lowerCAmelCase__			)							->				List[str]:
							return CLIPTokenizer.from_pretrained(self.tmpdirname      ,       pad_token="""!"""      ,       **lowerCAmelCase__			)
	def       __magic_name__    (    self      ,       **lowerCAmelCase__			)							->				Any:
							return CLIPTokenizerFast.from_pretrained(self.tmpdirname      ,       pad_token="""!"""      ,       **lowerCAmelCase__			)
	def       __magic_name__    (    self      ,       **lowerCAmelCase__			)							->				Optional[Any]:
							return OwlViTImageProcessor.from_pretrained(self.tmpdirname      ,       **lowerCAmelCase__			)
	def       __magic_name__    (    self			)							->				int:
							shutil.rmtree(self.tmpdirname			)
	def       __magic_name__    (    self			)							->				int:
							__magic_name__      :       str       						= [np.random.randint(2_55      ,       size=(3, 30, 4_00)      ,       dtype=np.uinta			)]

							__magic_name__      :       Any       						= [Image.fromarray(np.moveaxis(lowerCAmelCase__      ,       0      ,       -1			)			) for x in image_inputs]

							return image_inputs
	def       __magic_name__    (    self			)							->				Union[str, Any]:
							__magic_name__      :       Any       						= self.get_tokenizer()
							__magic_name__      :       str       						= self.get_rust_tokenizer()
							__magic_name__      :       Tuple       						= self.get_image_processor()

							__magic_name__      :       List[str]       						= OwlViTProcessor(tokenizer=lowerCAmelCase__      ,       image_processor=lowerCAmelCase__			)
							processor_slow.save_pretrained(self.tmpdirname			)
							__magic_name__      :       Optional[int]       						= OwlViTProcessor.from_pretrained(self.tmpdirname      ,       use_fast=lowerCAmelCase__			)

							__magic_name__      :       int       						= OwlViTProcessor(tokenizer=lowerCAmelCase__      ,       image_processor=lowerCAmelCase__			)
							processor_fast.save_pretrained(self.tmpdirname			)
							__magic_name__      :       List[Any]       						= OwlViTProcessor.from_pretrained(self.tmpdirname			)

							self.assertEqual(processor_slow.tokenizer.get_vocab()      ,       tokenizer_slow.get_vocab()			)
							self.assertEqual(processor_fast.tokenizer.get_vocab()      ,       tokenizer_fast.get_vocab()			)
							self.assertEqual(tokenizer_slow.get_vocab()      ,       tokenizer_fast.get_vocab()			)
							self.assertIsInstance(processor_slow.tokenizer      ,       lowerCAmelCase__			)
							self.assertIsInstance(processor_fast.tokenizer      ,       lowerCAmelCase__			)

							self.assertEqual(processor_slow.image_processor.to_json_string()      ,       image_processor.to_json_string()			)
							self.assertEqual(processor_fast.image_processor.to_json_string()      ,       image_processor.to_json_string()			)
							self.assertIsInstance(processor_slow.image_processor      ,       lowerCAmelCase__			)
							self.assertIsInstance(processor_fast.image_processor      ,       lowerCAmelCase__			)
	def       __magic_name__    (    self			)							->				Optional[int]:
							__magic_name__      :       Optional[Any]       						= OwlViTProcessor(tokenizer=self.get_tokenizer()      ,       image_processor=self.get_image_processor()			)
							processor.save_pretrained(self.tmpdirname			)

							__magic_name__      :       str       						= self.get_tokenizer(bos_token="""(BOS)"""      ,       eos_token="""(EOS)"""			)
							__magic_name__      :       Any       						= self.get_image_processor(do_normalize=lowerCAmelCase__			)

							__magic_name__      :       Tuple       						= OwlViTProcessor.from_pretrained(
							    self.tmpdirname      ,       bos_token="""(BOS)"""      ,       eos_token="""(EOS)"""      ,       do_normalize=lowerCAmelCase__			)

							self.assertEqual(processor.tokenizer.get_vocab()      ,       tokenizer_add_kwargs.get_vocab()			)
							self.assertIsInstance(processor.tokenizer      ,       lowerCAmelCase__			)

							self.assertEqual(processor.image_processor.to_json_string()      ,       image_processor_add_kwargs.to_json_string()			)
							self.assertIsInstance(processor.image_processor      ,       lowerCAmelCase__			)
	def       __magic_name__    (    self			)							->				Dict:
							__magic_name__      :       int       						= self.get_image_processor()
							__magic_name__      :       int       						= self.get_tokenizer()

							__magic_name__      :       Union[str, Any]       						= OwlViTProcessor(tokenizer=lowerCAmelCase__      ,       image_processor=lowerCAmelCase__			)

							__magic_name__      :       Dict       						= self.prepare_image_inputs()

							__magic_name__      :       Any       						= image_processor(lowerCAmelCase__      ,       return_tensors="""np"""			)
							__magic_name__      :       str       						= processor(images=lowerCAmelCase__      ,       return_tensors="""np"""			)

							for key in input_image_proc.keys():
													self.assertAlmostEqual(input_image_proc[key].sum()      ,       input_processor[key].sum()      ,       delta=1e-2			)
	def       __magic_name__    (    self			)							->				Tuple:
							__magic_name__      :       Union[str, Any]       						= self.get_image_processor()
							__magic_name__      :       int       						= self.get_tokenizer()

							__magic_name__      :       int       						= OwlViTProcessor(tokenizer=lowerCAmelCase__      ,       image_processor=lowerCAmelCase__			)

							__magic_name__      :       Optional[int]       						= """lower newer"""

							__magic_name__      :       Tuple       						= processor(text=lowerCAmelCase__      ,       return_tensors="""np"""			)

							__magic_name__      :       Optional[int]       						= tokenizer(lowerCAmelCase__      ,       return_tensors="""np"""			)

							for key in encoded_tok.keys():
													self.assertListEqual(encoded_tok[key][0].tolist()      ,       encoded_processor[key][0].tolist()			)
	def       __magic_name__    (    self			)							->				Tuple:
							__magic_name__      :       Tuple       						= self.get_image_processor()
							__magic_name__      :       Union[str, Any]       						= self.get_tokenizer()

							__magic_name__      :       List[str]       						= OwlViTProcessor(tokenizer=lowerCAmelCase__      ,       image_processor=lowerCAmelCase__			)

							__magic_name__      :       Any       						= """lower newer"""
							__magic_name__      :       Union[str, Any]       						= self.prepare_image_inputs()

							__magic_name__      :       int       						= processor(text=lowerCAmelCase__      ,       images=lowerCAmelCase__			)

							self.assertListEqual(list(inputs.keys()			)      ,       ["""input_ids""", """attention_mask""", """pixel_values"""]			)

							# test if it raises when no input is passed
							with pytest.raises(lowerCAmelCase__			):
													processor()
	def       __magic_name__    (    self			)							->				Optional[Any]:
							__magic_name__      :       Dict       						= """google/owlvit-base-patch32"""
							__magic_name__      :       int       						= OwlViTProcessor.from_pretrained(lowerCAmelCase__			)

							__magic_name__      :       List[Any]       						= ["""cat""", """nasa badge"""]
							__magic_name__      :       Any       						= processor(text=lowerCAmelCase__			)

							__magic_name__      :       Dict       						= 16
							self.assertListEqual(list(inputs.keys()			)      ,       ["""input_ids""", """attention_mask"""]			)
							self.assertEqual(inputs["""input_ids"""].shape      ,       (2, seq_length)			)

							# test if it raises when no input is passed
							with pytest.raises(lowerCAmelCase__			):
													processor()
	def       __magic_name__    (    self			)							->				List[Any]:
							__magic_name__      :       List[str]       						= """google/owlvit-base-patch32"""
							__magic_name__      :       Optional[Any]       						= OwlViTProcessor.from_pretrained(lowerCAmelCase__			)

							__magic_name__      :       Tuple       						= [["""cat""", """nasa badge"""], ["""person"""]]
							__magic_name__      :       Tuple       						= processor(text=lowerCAmelCase__			)

							__magic_name__      :       str       						= 16
							__magic_name__      :       str       						= len(lowerCAmelCase__			)
							__magic_name__      :       int       						= max([len(lowerCAmelCase__			) for texts in input_texts]			)

							self.assertListEqual(list(inputs.keys()			)      ,       ["""input_ids""", """attention_mask"""]			)
							self.assertEqual(inputs["""input_ids"""].shape      ,       (batch_size * num_max_text_queries, seq_length)			)

							# test if it raises when no input is passed
							with pytest.raises(lowerCAmelCase__			):
													processor()
	def       __magic_name__    (    self			)							->				Any:
							__magic_name__      :       Optional[int]       						= """google/owlvit-base-patch32"""
							__magic_name__      :       Any       						= OwlViTProcessor.from_pretrained(lowerCAmelCase__			)

							__magic_name__      :       str       						= ["""cat""", """nasa badge"""]
							__magic_name__      :       List[str]       						= processor(text=lowerCAmelCase__			)

							__magic_name__      :       List[Any]       						= 16
							__magic_name__      :       Any       						= inputs["""input_ids"""]
							__magic_name__      :       Optional[Any]       						= [
							    [4_94_06, 23_68, 4_94_07, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
							    [4_94_06, 68_41, 1_13_01, 4_94_07, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
							]

							self.assertListEqual(list(inputs.keys()			)      ,       ["""input_ids""", """attention_mask"""]			)
							self.assertEqual(inputs["""input_ids"""].shape      ,       (2, seq_length)			)
							self.assertListEqual(list(input_ids[0]			)      ,       predicted_ids[0]			)
							self.assertListEqual(list(input_ids[1]			)      ,       predicted_ids[1]			)
	def       __magic_name__    (    self			)							->				Tuple:
							__magic_name__      :       List[str]       						= self.get_image_processor()
							__magic_name__      :       Dict       						= self.get_tokenizer()

							__magic_name__      :       Tuple       						= OwlViTProcessor(tokenizer=lowerCAmelCase__      ,       image_processor=lowerCAmelCase__			)

							__magic_name__      :       Tuple       						= self.prepare_image_inputs()
							__magic_name__      :       List[Any]       						= self.prepare_image_inputs()

							__magic_name__      :       List[str]       						= processor(images=lowerCAmelCase__      ,       query_images=lowerCAmelCase__			)

							self.assertListEqual(list(inputs.keys()			)      ,       ["""query_pixel_values""", """pixel_values"""]			)

							# test if it raises when no input is passed
							with pytest.raises(lowerCAmelCase__			):
													processor()







	def       __magic_name__    (    self			)							->				Any:
							__magic_name__      :       Optional[Any]       						= self.get_image_processor()
							__magic_name__      :       List[Any]       						= self.get_tokenizer()

							__magic_name__      :       Tuple       						= OwlViTProcessor(tokenizer=lowerCAmelCase__      ,       image_processor=lowerCAmelCase__			)

							__magic_name__      :       Union[str, Any]       						= [[1, 4, 5, 8, 1, 0, 8], [3, 4, 3, 1, 1, 8, 9]]

							__magic_name__      :       Optional[Any]       						= processor.batch_decode(lowerCAmelCase__			)
							__magic_name__      :       Optional[int]       						= tokenizer.batch_decode(lowerCAmelCase__			)

							self.assertListEqual(lowerCAmelCase__      ,       lowerCAmelCase__			)
