
import sacrebleu as scb
from packaging import version

import datasets


UpperCAmelCase     				=			'''\
@inproceedings{post-2018-call,
    title = "A Call for Clarity in Reporting {BLEU} Scores",
    author = "Post, Matt",
    booktitle = "Proceedings of the Third Conference on Machine Translation: Research Papers",
    month = oct,
    year = "2018",
    address = "Belgium, Brussels",
    publisher = "Association for Computational Linguistics",
    url = "https://www.aclweb.org/anthology/W18-6319",
    pages = "186--191",
}
'''

UpperCAmelCase     				=			'''\
SacreBLEU provides hassle-free computation of shareable, comparable, and reproducible BLEU scores.
Inspired by Rico Sennrich\'s `multi-bleu-detok.perl`, it produces the official WMT scores but works with plain text.
It also knows all the standard test sets and handles downloading, processing, and tokenization for you.

See the [README.md] file at https://github.com/mjpost/sacreBLEU for more information.
'''

UpperCAmelCase     				=			'''
Produces BLEU scores along with its sufficient statistics
from a source against one or more references.

Args:
    predictions (`list` of `str`): list of translations to score. Each translation should be tokenized into a list of tokens.
    references (`list` of `list` of `str`): A list of lists of references. The contents of the first sub-list are the references for the first prediction, the contents of the second sub-list are for the second prediction, etc. Note that there must be the same number of references for each prediction (i.e. all sub-lists must be of the same length).
    smooth_method (`str`): The smoothing method to use, defaults to `\'exp\'`. Possible values are:
        - `\'none\'`: no smoothing
        - `\'floor\'`: increment zero counts
        - `\'add-k\'`: increment num/denom by k for n>1
        - `\'exp\'`: exponential decay
    smooth_value (`float`): The smoothing value. Only valid when `smooth_method=\'floor\'` (in which case `smooth_value` defaults to `0.1`) or `smooth_method=\'add-k\'` (in which case `smooth_value` defaults to `1`).
    tokenize (`str`): Tokenization method to use for BLEU. If not provided, defaults to `\'zh\'` for Chinese, `\'ja-mecab\'` for Japanese and `\'13a\'` (mteval) otherwise. Possible values are:
        - `\'none\'`: No tokenization.
        - `\'zh\'`: Chinese tokenization.
        - `\'13a\'`: mimics the `mteval-v13a` script from Moses.
        - `\'intl\'`: International tokenization, mimics the `mteval-v14` script from Moses
        - `\'char\'`: Language-agnostic character-level tokenization.
        - `\'ja-mecab\'`: Japanese tokenization. Uses the [MeCab tokenizer](https://pypi.org/project/mecab-python3).
    lowercase (`bool`): If `True`, lowercases the input, enabling case-insensitivity. Defaults to `False`.
    force (`bool`): If `True`, insists that your tokenized input is actually detokenized. Defaults to `False`.
    use_effective_order (`bool`): If `True`, stops including n-gram orders for which precision is 0. This should be `True`, if sentence-level BLEU will be computed. Defaults to `False`.

Returns:
    \'score\': BLEU score,
    \'counts\': Counts,
    \'totals\': Totals,
    \'precisions\': Precisions,
    \'bp\': Brevity penalty,
    \'sys_len\': predictions length,
    \'ref_len\': reference length,

Examples:

    Example 1:
        >>> predictions = ["hello there general kenobi", "foo bar foobar"]
        >>> references = [["hello there general kenobi", "hello there !"], ["foo bar foobar", "foo bar foobar"]]
        >>> sacrebleu = datasets.load_metric("sacrebleu")
        >>> results = sacrebleu.compute(predictions=predictions, references=references)
        >>> print(list(results.keys()))
        [\'score\', \'counts\', \'totals\', \'precisions\', \'bp\', \'sys_len\', \'ref_len\']
        >>> print(round(results["score"], 1))
        100.0

    Example 2:
        >>> predictions = ["hello there general kenobi",
        ...                 "on our way to ankh morpork"]
        >>> references = [["hello there general kenobi", "hello there !"],
        ...                 ["goodbye ankh morpork", "ankh morpork"]]
        >>> sacrebleu = datasets.load_metric("sacrebleu")
        >>> results = sacrebleu.compute(predictions=predictions,
        ...                             references=references)
        >>> print(list(results.keys()))
        [\'score\', \'counts\', \'totals\', \'precisions\', \'bp\', \'sys_len\', \'ref_len\']
        >>> print(round(results["score"], 1))
        39.8
'''







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




	'''simple docstring'''


	def      SCREAMING_SNAKE_CASE__						(	self	):
		if version.parse(scb.__version__	) < version.parse('1.4.12'	):
			raise ImportWarning(
			    'To use `sacrebleu`, the module `sacrebleu>=1.4.12` is required, and the current version of `sacrebleu` doesn\'t match this condition.\n'
			    'You can install it with `pip install "sacrebleu>=1.4.12"`.'	)
		return datasets.MetricInfo(
		    description=_DESCRIPTION		,   citation=_CITATION		,   homepage='https://github.com/mjpost/sacreBLEU'		,   inputs_description=_KWARGS_DESCRIPTION		,   features=datasets.Features(
		        {
		            'predictions': datasets.Value('string'		,   id='sequence'	),
		            'references': datasets.Sequence(datasets.Value('string'		,   id='sequence'	)		,   id='references'	),
		        }	)		,   codebase_urls=['https://github.com/mjpost/sacreBLEU']		,   reference_urls=[
		        'https://github.com/mjpost/sacreBLEU',
		        'https://en.wikipedia.org/wiki/BLEU',
		        'https://towardsdatascience.com/evaluating-text-output-in-nlp-bleu-at-your-own-risk-e8609665a213',
		    ]		,   )





	def      SCREAMING_SNAKE_CASE__						(	self		,   snake_case		,   snake_case		,   snake_case="exp"		,   snake_case=None		,   snake_case=False		,   snake_case=False		,   snake_case=None		,   snake_case=False		,   ):
		lowercase					  =   len(references[0]	)
		if any(len(snake_case	) != references_per_prediction for refs in references	):
			raise ValueError('Sacrebleu requires the same number of references for each prediction'	)
		lowercase					  =   [[refs[i] for refs in references] for i in range(snake_case	)]
		lowercase					  =   scb.corpus_bleu(
		    snake_case		,   snake_case		,   smooth_method=snake_case		,   smooth_value=snake_case		,   force=snake_case		,   lowercase=snake_case		,   use_effective_order=snake_case		,   **({'tokenize': tokenize} if tokenize else {})		,   )
		lowercase					  =   {
		    'score': output.score,
		    'counts': output.counts,
		    'totals': output.totals,
		    'precisions': output.precisions,
		    'bp': output.bp,
		    'sys_len': output.sys_len,
		    'ref_len': output.ref_len,
		}
		return output_dict



import unittest

import numpy as np

from transformers.testing_utils import require_torch, require_vision
from transformers.utils import is_torch_available, is_vision_available

from ...test_image_processing_common import ImageProcessingSavingTestMixin, prepare_video_inputs


if is_torch_available():
	import torch

if is_vision_available():
	from PIL import Image

	from transformers import VideoMAEImageProcessor







class 	A_       (      unittest.TestCase       ):




	'''simple docstring'''


	def __init__(	self		,   snake_case		,   snake_case=7		,   snake_case=3		,   snake_case=10		,   snake_case=18		,   snake_case=30		,   snake_case=400		,   snake_case=True		,   snake_case=None		,   snake_case=True		,   snake_case=[0.5, 0.5, 0.5]		,   snake_case=[0.5, 0.5, 0.5]		,   snake_case=None		,   ):
		lowercase					  =   size if size is not None else {'shortest_edge': 18}
		lowercase					  =   crop_size if crop_size is not None else {'height': 18, 'width': 18}

		lowercase					  =   parent
		lowercase					  =   batch_size
		lowercase					  =   num_channels
		lowercase					  =   num_frames
		lowercase					  =   image_size
		lowercase					  =   min_resolution
		lowercase					  =   max_resolution
		lowercase					  =   do_resize
		lowercase					  =   size
		lowercase					  =   do_normalize
		lowercase					  =   image_mean
		lowercase					  =   image_std
		lowercase					  =   crop_size





	def      SCREAMING_SNAKE_CASE__						(	self	):
		return {
		    "image_mean": self.image_mean,
		    "image_std": self.image_std,
		    "do_normalize": self.do_normalize,
		    "do_resize": self.do_resize,
		    "size": self.size,
		    "crop_size": self.crop_size,
		}







@require_torch
@require_vision
class 	A_       (      __lowerCamelCase				,     unittest.TestCase       ):




	'''simple docstring'''


	_UpperCamelCase					:					List[str]										=	VideoMAEImageProcessor if is_vision_available() else None





	def      SCREAMING_SNAKE_CASE__						(	self	):
		lowercase					  =   VideoMAEImageProcessingTester(self	)





	@property
	def      SCREAMING_SNAKE_CASE__						(	self	):
		return self.image_processor_tester.prepare_image_processor_dict()





	def      SCREAMING_SNAKE_CASE__						(	self	):
		lowercase					  =   self.image_processing_class(**self.image_processor_dict	)
		self.assertTrue(hasattr(snake_case		,   'image_mean'	)	)
		self.assertTrue(hasattr(snake_case		,   'image_std'	)	)
		self.assertTrue(hasattr(snake_case		,   'do_normalize'	)	)
		self.assertTrue(hasattr(snake_case		,   'do_resize'	)	)
		self.assertTrue(hasattr(snake_case		,   'do_center_crop'	)	)
		self.assertTrue(hasattr(snake_case		,   'size'	)	)





	def      SCREAMING_SNAKE_CASE__						(	self	):
		lowercase					  =   self.image_processing_class.from_dict(self.image_processor_dict	)
		self.assertEqual(image_processor.size		,   {'shortest_edge': 18}	)
		self.assertEqual(image_processor.crop_size		,   {'height': 18, 'width': 18}	)

		lowercase					  =   self.image_processing_class.from_dict(self.image_processor_dict		,   size=42		,   crop_size=84	)
		self.assertEqual(image_processor.size		,   {'shortest_edge': 42}	)
		self.assertEqual(image_processor.crop_size		,   {'height': 84, 'width': 84}	)





	def      SCREAMING_SNAKE_CASE__						(	self	):
		pass





	def      SCREAMING_SNAKE_CASE__						(	self	):
		# Initialize image_processing
		lowercase					  =   self.image_processing_class(**self.image_processor_dict	)
		# create random PIL videos
		lowercase					  =   prepare_video_inputs(self.image_processor_tester		,   equal_resolution=snake_case	)
		for video in video_inputs:
			self.assertIsInstance(snake_case		,   snake_case	)
			self.assertIsInstance(video[0]		,   Image.Image	)

		# Test not batched input
		lowercase					  =   image_processing(video_inputs[0]		,   return_tensors='pt'	).pixel_values
		self.assertEqual(
		    encoded_videos.shape		,   (
		        1,
		        self.image_processor_tester.num_frames,
		        self.image_processor_tester.num_channels,
		        self.image_processor_tester.crop_size['height'],
		        self.image_processor_tester.crop_size['width'],
		    )		,   )

		# Test batched
		lowercase					  =   image_processing(snake_case		,   return_tensors='pt'	).pixel_values
		self.assertEqual(
		    encoded_videos.shape		,   (
		        self.image_processor_tester.batch_size,
		        self.image_processor_tester.num_frames,
		        self.image_processor_tester.num_channels,
		        self.image_processor_tester.crop_size['height'],
		        self.image_processor_tester.crop_size['width'],
		    )		,   )





	def      SCREAMING_SNAKE_CASE__						(	self	):
		# Initialize image_processing
		lowercase					  =   self.image_processing_class(**self.image_processor_dict	)
		# create random numpy tensors
		lowercase					  =   prepare_video_inputs(self.image_processor_tester		,   equal_resolution=snake_case		,   numpify=snake_case	)
		for video in video_inputs:
			self.assertIsInstance(snake_case		,   snake_case	)
			self.assertIsInstance(video[0]		,   np.ndarray	)

		# Test not batched input
		lowercase					  =   image_processing(video_inputs[0]		,   return_tensors='pt'	).pixel_values
		self.assertEqual(
		    encoded_videos.shape		,   (
		        1,
		        self.image_processor_tester.num_frames,
		        self.image_processor_tester.num_channels,
		        self.image_processor_tester.crop_size['height'],
		        self.image_processor_tester.crop_size['width'],
		    )		,   )

		# Test batched
		lowercase					  =   image_processing(snake_case		,   return_tensors='pt'	).pixel_values
		self.assertEqual(
		    encoded_videos.shape		,   (
		        self.image_processor_tester.batch_size,
		        self.image_processor_tester.num_frames,
		        self.image_processor_tester.num_channels,
		        self.image_processor_tester.crop_size['height'],
		        self.image_processor_tester.crop_size['width'],
		    )		,   )





	def      SCREAMING_SNAKE_CASE__						(	self	):
		# Initialize image_processing
		lowercase					  =   self.image_processing_class(**self.image_processor_dict	)
		# create random PyTorch tensors
		lowercase					  =   prepare_video_inputs(self.image_processor_tester		,   equal_resolution=snake_case		,   torchify=snake_case	)
		for video in video_inputs:
			self.assertIsInstance(snake_case		,   snake_case	)
			self.assertIsInstance(video[0]		,   torch.Tensor	)

		# Test not batched input
		lowercase					  =   image_processing(video_inputs[0]		,   return_tensors='pt'	).pixel_values
		self.assertEqual(
		    encoded_videos.shape		,   (
		        1,
		        self.image_processor_tester.num_frames,
		        self.image_processor_tester.num_channels,
		        self.image_processor_tester.crop_size['height'],
		        self.image_processor_tester.crop_size['width'],
		    )		,   )

		# Test batched
		lowercase					  =   image_processing(snake_case		,   return_tensors='pt'	).pixel_values
		self.assertEqual(
		    encoded_videos.shape		,   (
		        self.image_processor_tester.batch_size,
		        self.image_processor_tester.num_frames,
		        self.image_processor_tester.num_channels,
		        self.image_processor_tester.crop_size['height'],
		        self.image_processor_tester.crop_size['width'],
		    )		,   )

