






'''simple docstring'''



import sacrebleu as scb
from packaging import version
from sacrebleu import TER

import datasets


SCREAMING_SNAKE_CASE							=						'\\n@inproceedings{snover-etal-2006-study,\n    title = "A Study of Translation Edit Rate with Targeted Human Annotation",\n    author = "Snover, Matthew  and\n      Dorr, Bonnie  and\n      Schwartz, Rich  and\n      Micciulla, Linnea  and\n      Makhoul, John",\n    booktitle = "Proceedings of the 7th Conference of the Association for Machine Translation in the Americas: Technical Papers",\n    month = aug # " 8-12",\n    year = "2006",\n    address = "Cambridge, Massachusetts, USA",\n    publisher = "Association for Machine Translation in the Americas",\n    url = "https://aclanthology.org/2006.amta-papers.25",\n    pages = "223--231",\n}\n@inproceedings{post-2018-call,\n    title = "A Call for Clarity in Reporting {BLEU} Scores",\n    author = "Post, Matt",\n    booktitle = "Proceedings of the Third Conference on Machine Translation: Research Papers",\n    month = oct,\n    year = "2018",\n    address = "Belgium, Brussels",\n    publisher = "Association for Computational Linguistics",\n    url = "https://www.aclweb.org/anthology/W18-6319",\n    pages = "186--191",\n}\n'

SCREAMING_SNAKE_CASE							=						'\\nTER (Translation Edit Rate, also called Translation Error Rate) is a metric to quantify the edit operations that a\nhypothesis requires to match a reference translation. We use the implementation that is already present in sacrebleu\n(https://github.com/mjpost/sacreBLEU#ter), which in turn is inspired by the TERCOM implementation, which can be found\nhere: https://github.com/jhclark/tercom.\n\nThe implementation here is slightly different from sacrebleu in terms of the required input format. The length of\nthe references and hypotheses lists need to be the same, so you may need to transpose your references compared to\nsacrebleu\'s required input format. See https://github.com/huggingface/datasets/issues/3154#issuecomment-950746534\n\nSee the README.md file at https://github.com/mjpost/sacreBLEU#ter for more information.\n'

SCREAMING_SNAKE_CASE							=						'\nProduces TER scores alongside the number of edits and reference length.\n\nArgs:\n    predictions (list of str): The system stream (a sequence of segments).\n    references (list of list of str): A list of one or more reference streams (each a sequence of segments).\n    normalized (boolean): If `True`, applies basic tokenization and normalization to sentences. Defaults to `False`.\n    ignore_punct (boolean): If `True`, applies basic tokenization and normalization to sentences. Defaults to `False`.\n    support_zh_ja_chars (boolean): If `True`, tokenization/normalization supports processing of Chinese characters,\n                                    as well as Japanese Kanji, Hiragana, Katakana, and Phonetic Extensions of Katakana.\n                                    Only applies if `normalized = True`. Defaults to `False`.\n    case_sensitive (boolean): If `False`, makes all predictions and references lowercase to ignore differences in case. Defaults to `False`.\n\nReturns:\n    \'score\' (float): TER score (num_edits / sum_ref_lengths * 100)\n    \'num_edits\' (int): The cumulative number of edits\n    \'ref_length\' (float): The cumulative average reference length\n\nExamples:\n    Example 1:\n        >>> predictions = ["does this sentence match??",\n        ...                     "what about this sentence?",\n        ...                     "What did the TER metric user say to the developer?"]\n        >>> references = [["does this sentence match", "does this sentence match!?!"],\n        ...             ["wHaT aBoUt ThIs SeNtEnCe?", "wHaT aBoUt ThIs SeNtEnCe?"],\n        ...             ["Your jokes are...", "...TERrible"]]\n        >>> ter = datasets.load_metric("ter")\n        >>> results = ter.compute(predictions=predictions,\n        ...                         references=references,\n        ...                         case_sensitive=True)\n        >>> print(results)\n        {\'score\': 150.0, \'num_edits\': 15, \'ref_length\': 10.0}\n\n    Example 2:\n        >>> predictions = ["does this sentence match??",\n        ...                     "what about this sentence?"]\n        >>> references = [["does this sentence match", "does this sentence match!?!"],\n        ...             ["wHaT aBoUt ThIs SeNtEnCe?", "wHaT aBoUt ThIs SeNtEnCe?"]]\n        >>> ter = datasets.load_metric("ter")\n        >>> results = ter.compute(predictions=predictions,\n        ...                         references=references,\n        ...                         case_sensitive=True)\n        >>> print(results)\n        {\'score\': 62.5, \'num_edits\': 5, \'ref_length\': 8.0}\n\n    Example 3:\n        >>> predictions = ["does this sentence match??",\n        ...                     "what about this sentence?"]\n        >>> references = [["does this sentence match", "does this sentence match!?!"],\n        ...             ["wHaT aBoUt ThIs SeNtEnCe?", "wHaT aBoUt ThIs SeNtEnCe?"]]\n        >>> ter = datasets.load_metric("ter")\n        >>> results = ter.compute(predictions=predictions,\n        ...                         references=references,\n        ...                         normalized=True,\n        ...                         case_sensitive=True)\n        >>> print(results)\n        {\'score\': 57.14285714285714, \'num_edits\': 6, \'ref_length\': 10.5}\n\n    Example 4:\n        >>> predictions = ["does this sentence match??",\n        ...                     "what about this sentence?"]\n        >>> references = [["does this sentence match", "does this sentence match!?!"],\n        ...             ["wHaT aBoUt ThIs SeNtEnCe?", "wHaT aBoUt ThIs SeNtEnCe?"]]\n        >>> ter = datasets.load_metric("ter")\n        >>> results = ter.compute(predictions=predictions,\n        ...                         references=references,\n        ...                         ignore_punct=True,\n        ...                         case_sensitive=False)\n        >>> print(results)\n        {\'score\': 0.0, \'num_edits\': 0, \'ref_length\': 8.0}\n\n    Example 5:\n        >>> predictions = ["does this sentence match??",\n        ...                    "what about this sentence?",\n        ...                    "What did the TER metric user say to the developer?"]\n        >>> references = [["does this sentence match", "does this sentence match!?!"],\n        ...             ["wHaT aBoUt ThIs SeNtEnCe?", "wHaT aBoUt ThIs SeNtEnCe?"],\n        ...             ["Your jokes are...", "...TERrible"]]\n        >>> ter = datasets.load_metric("ter")\n        >>> results = ter.compute(predictions=predictions,\n        ...                         references=references,\n        ...                         ignore_punct=True,\n        ...                         case_sensitive=False)\n        >>> print(results)\n        {\'score\': 100.0, \'num_edits\': 10, \'ref_length\': 10.0}\n'
@datasets.utils.file_utils.add_start_docstrings(_DESCRIPTION					,							_KWARGS_DESCRIPTION					)
class UpperCAmelCase_       (     datasets.Metric					):




      """simple docstring"""




      def 					A__   (			self      :		Tuple		)	->	Dict:


       '''simple docstring'''

       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='''http://www.cs.umd.edu/~snover/tercom/'''  ,   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#ter''']  ,   reference_urls=[
               '''https://github.com/jhclark/tercom''',
           ]  ,   )






      def 					A__   (			self      :		Optional[Any]  ,   UpperCAmelCase      :		List[Any]  ,   UpperCAmelCase      :		str  ,   UpperCAmelCase      :		bool = False  ,   UpperCAmelCase      :		bool = False  ,   UpperCAmelCase      :		bool = False  ,   UpperCAmelCase      :		bool = False  ,   )	->	Union[str, Any]:


       '''simple docstring'''

       lowercase       : Union[str, Any]						=len(references[0]		)
       if any(len(UpperCAmelCase		) != references_per_prediction for refs in references		):
        raise ValueError('''Sacrebleu requires the same number of references for each prediction'''		)
       lowercase       : List[Any]						=[[refs[i] for refs in references] for i in range(UpperCAmelCase		)]

       lowercase       : int						=TER(
           normalized=UpperCAmelCase  ,   no_punct=UpperCAmelCase  ,   asian_support=UpperCAmelCase  ,   case_sensitive=UpperCAmelCase  ,   )
       lowercase       : Union[str, Any]						=sb_ter.corpus_score(UpperCAmelCase  ,   UpperCAmelCase		)

       return {"score": output.score, "num_edits": output.num_edits, "ref_length": output.ref_length}





"""simple docstring"""





from typing import Dict, List, Optional, Union

import numpy as np

from ...image_processing_utils import BaseImageProcessor, BatchFeature, get_size_dict
from ...image_transforms import (
    center_crop,
    get_resize_output_image_size,
    normalize,
    rescale,
    resize,
    to_channel_dimension_format,
)
from ...image_utils import (
    IMAGENET_DEFAULT_MEAN,
    IMAGENET_DEFAULT_STD,
    ChannelDimension,
    ImageInput,
    PILImageResampling,
    is_batched,
    to_numpy_array,
    valid_images,
)
from ...utils import TensorType, logging


_a								=						logging.get_logger(__name__)






class        _UpperCAmelCase(       lowerCamelCase      ):
		lowercase__         = ['pixel_values']



		def __init__(   self							,						__a = True							,						__a = None							,						__a = PILImageResampling.BICUBIC							,						__a = True							,						__a = True							,						__a = 1 / 2_55							,						__a = None							,						__a = True							,						__a = None							,						__a = None							,						**__a							,						)			->			None:






							'''simple docstring'''
							super().__init__(**__a)
							_UpperCamelCase									=    size if size is not None else {'''height''': 2_24, '''width''': 2_24}
							_UpperCamelCase									=    get_size_dict(__a)
							_UpperCamelCase									=    crop_size if crop_size is not None else {'''height''': 2_24, '''width''': 2_24}
							_UpperCamelCase									=    get_size_dict(__a							,						default_to_square=__a							,						param_name='''crop_size''')

							_UpperCamelCase									=    do_resize
							_UpperCamelCase									=    do_rescale
							_UpperCamelCase									=    do_normalize
							_UpperCamelCase									=    do_center_crop
							_UpperCamelCase									=    crop_size
							_UpperCamelCase									=    size
							_UpperCamelCase									=    resample
							_UpperCamelCase									=    rescale_factor
							_UpperCamelCase									=    image_mean if image_mean is not None else IMAGENET_DEFAULT_MEAN
							_UpperCamelCase									=    image_std if image_std is not None else IMAGENET_DEFAULT_STD



		def 		UpperCAmelCase				(   self							,						__a							,						__a							,						__a = PILImageResampling.BILINEAR							,						__a = None							,						**__a							,						)			->			np.ndarray:






							'''simple docstring'''
							_UpperCamelCase									=    get_size_dict(__a)

							if "shortest_edge" in size:
												_UpperCamelCase									=    get_resize_output_image_size(__a							,						size=size['''shortest_edge''']							,						default_to_square=__a)
												# size = get_resize_output_image_size(image, size["shortest_edge"], size["longest_edge"])
							elif "height" in size and "width" in size:
												_UpperCamelCase									=    (size['''height'''], size['''width'''])
							else:
												raise ValueError(F'''Size must contain \'height\' and \'width\' keys or \'shortest_edge\' key. Got {size.keys()}''')
							return resize(__a							,						size=__a							,						resample=__a							,						data_format=__a							,						**__a)



		def 		UpperCAmelCase				(   self							,						__a							,						__a							,						__a = None							,						**__a							,						)			->			np.ndarray:






							'''simple docstring'''
							_UpperCamelCase									=    get_size_dict(__a)
							if "height" not in size or "width" not in size:
												raise ValueError(F'''The `size` parameter must contain the keys (height, width). Got {size.keys()}''')
							return center_crop(__a							,						size=(size['''height'''], size['''width'''])							,						data_format=__a							,						**__a)



		def 		UpperCAmelCase				(   self							,						__a							,						__a							,						__a = None							,						**__a)			->			np.ndarray:






							'''simple docstring'''
							return rescale(__a							,						scale=__a							,						data_format=__a							,						**__a)



		def 		UpperCAmelCase				(   self							,						__a							,						__a							,						__a							,						__a = None							,						**__a							,						)			->			np.ndarray:






							'''simple docstring'''
							return normalize(__a							,						mean=__a							,						std=__a							,						data_format=__a							,						**__a)




		def 		UpperCAmelCase				(   self							,						__a							,						__a = None							,						__a = None							,						__a = None							,						__a = None							,						__a = None							,						__a = None							,						__a = None							,						__a = None							,						__a = None							,						__a = None							,						__a = None							,						__a = ChannelDimension.FIRST							,						**__a							,						)			->			BatchFeature:






							'''simple docstring'''
							_UpperCamelCase									=    do_resize if do_resize is not None else self.do_resize
							_UpperCamelCase									=    do_rescale if do_rescale is not None else self.do_rescale
							_UpperCamelCase									=    do_normalize if do_normalize is not None else self.do_normalize
							_UpperCamelCase									=    do_center_crop if do_center_crop is not None else self.do_center_crop
							_UpperCamelCase									=    crop_size if crop_size is not None else self.crop_size
							_UpperCamelCase									=    get_size_dict(__a							,						param_name='''crop_size'''							,						default_to_square=__a)
							_UpperCamelCase									=    resample if resample is not None else self.resample
							_UpperCamelCase									=    rescale_factor if rescale_factor is not None else self.rescale_factor
							_UpperCamelCase									=    image_mean if image_mean is not None else self.image_mean
							_UpperCamelCase									=    image_std if image_std is not None else self.image_std

							_UpperCamelCase									=    size if size is not None else self.size
							_UpperCamelCase									=    get_size_dict(__a)

							if not is_batched(__a):
												_UpperCamelCase									=    [images]

							if not valid_images(__a):
												raise ValueError(
												    '''Invalid image type. Must be of type PIL.Image.Image, numpy.ndarray, '''
												    '''torch.Tensor, tf.Tensor or jax.ndarray.''')

							if do_resize and size is None:
												raise ValueError('''Size must be specified if do_resize is True.''')

							if do_center_crop and crop_size is None:
												raise ValueError('''Crop size must be specified if do_center_crop is True.''')

							if do_rescale and rescale_factor is None:
												raise ValueError('''Rescale factor must be specified if do_rescale is True.''')

							# All transformations expect numpy arrays.
							_UpperCamelCase									=    [to_numpy_array(__a) for image in images]

							if do_resize:
												_UpperCamelCase									=    [self.resize(image=__a							,						size=__a							,						resample=__a) for image in images]

							if do_center_crop:
												_UpperCamelCase									=    [self.center_crop(image=__a							,						size=__a) for image in images]

							if do_rescale:
												_UpperCamelCase									=    [self.rescale(image=__a							,						scale=__a) for image in images]

							if do_normalize:
												_UpperCamelCase									=    [self.normalize(image=__a							,						mean=__a							,						std=__a) for image in images]

							_UpperCamelCase									=    [to_channel_dimension_format(__a							,						__a) for image in images]

							_UpperCamelCase									=    {'''pixel_values''': images}
							return BatchFeature(data=__a							,						tensor_type=__a)






