


import comet  # From: unbabel-comet
import torch

import datasets


snake_case__			 =						datasets.logging.get_logger(__name__)

snake_case__			 =						"\\n@inproceedings{rei-EtAl:2020:WMT,\n   author    = {Rei, Ricardo  and  Stewart, Craig  and  Farinha, Ana C  and  Lavie, Alon},\n   title     = {Unbabel's Participation in the WMT20 Metrics Shared Task},\n   booktitle      = {Proceedings of the Fifth Conference on Machine Translation},\n   month          = {November},\n   year           = {2020},\n   address        = {Online},\n   publisher      = {Association for Computational Linguistics},\n   pages     = {909--918},\n}\n@inproceedings{rei-etal-2020-comet,\n   title = \"{COMET}: A Neural Framework for {MT} Evaluation\",\n   author = \"Rei, Ricardo  and\n      Stewart, Craig  and\n      Farinha, Ana C  and\n      Lavie, Alon\",\n   booktitle = \"Proceedings of the 2020 Conference on Empirical Methods in Natural Language Processing (EMNLP)\",\n   month = nov,\n   year = \"2020\",\n   address = \"Online\",\n   publisher = \"Association for Computational Linguistics\",\n   url = \"https://www.aclweb.org/anthology/2020.emnlp-main.213\",\n   pages = \"2685--2702\",\n}\n"

snake_case__			 =						"\\nCrosslingual Optimized Metric for Evaluation of Translation (COMET) is an open-source framework used to train Machine Translation metrics that achieve high levels of correlation with different types of human judgments (HTER, DA's or MQM).\nWith the release of the framework the authors also released fully trained models that were used to compete in the WMT20 Metrics Shared Task achieving SOTA in that years competition.\n\nSee the [README.md] file at https://unbabel.github.io/COMET/html/models.html for more information.\n"

snake_case__			 =						"\nCOMET score.\n\nArgs:\n\n`sources` (list of str): Source sentences\n`predictions` (list of str): candidate translations\n`references` (list of str): reference translations\n`cuda` (bool): If set to True, runs COMET using GPU\n`show_progress` (bool): Shows progress\n`model`: COMET model to be used. Will default to `wmt-large-da-estimator-1719` if None.\n\nReturns:\n    `samples`: List of dictionaries with `src`, `mt`, `ref` and `score`.\n    `scores`: List of scores.\n\nExamples:\n\n    >>> comet_metric = datasets.load_metric('comet')\n    >>> # comet_metric = load_metric('comet', 'wmt20-comet-da')  # you can also choose which model to use\n    >>> source = [\"Dem Feuer konnte Einhalt geboten werden\", \"Schulen und Kindergärten wurden eröffnet.\"]\n    >>> hypothesis = [\"The fire could be stopped\", \"Schools and kindergartens were open\"]\n    >>> reference = [\"They were able to control the fire.\", \"Schools and kindergartens opened\"]\n    >>> results = comet_metric.compute(predictions=hypothesis, references=reference, sources=source)\n    >>> print([round(v, 2) for v in results[\"scores\"]])\n    [0.19, 0.92]\n"


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




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






									"""simple docstring"""






									return datasets.MetricInfo(
									    description=_DESCRIPTION   ,     citation=_CITATION   ,     homepage="https://unbabel.github.io/COMET/html/index.html"   ,     inputs_description=_KWARGS_DESCRIPTION   ,     features=datasets.Features(
									        {
									            "sources": datasets.Value("string"   ,     id="sequence"						),
									            "predictions": datasets.Value("string"   ,     id="sequence"						),
									            "references": datasets.Value("string"   ,     id="sequence"						),
									        }						)   ,     codebase_urls=["https://github.com/Unbabel/COMET"]   ,     reference_urls=[
									        "https://github.com/Unbabel/COMET",
									        "https://www.aclweb.org/anthology/2020.emnlp-main.213/",
									        "http://www.statmt.org/wmt20/pdf/2020.wmt-1.101.pdf6",
									    ]   ,     )




						def  _snake_case					(	self						:   Tuple   ,     __A						:   List[str]						)      ->Optional[Any]:






									"""simple docstring"""






									if self.config_name == "default":
												a__ :int						  =    comet.load_from_checkpoint(comet.download_model("wmt20-comet-da"						)						)
									else:
												a__ :Any						  =    comet.load_from_checkpoint(comet.download_model(self.config_name						)						)






						def  _snake_case					(	self						:   int   ,     __A						:   Optional[Any]   ,     __A						:   Optional[Any]   ,     __A						:   int   ,     __A						:   Any=None   ,     __A						:   Any=False						)      ->Tuple:






									"""simple docstring"""






									if gpus is None:
												a__ :Tuple						  =    1 if torch.cuda.is_available() else 0
									a__ :int						  =    {"src": sources, "mt": predictions, "ref": references}
									a__ :Optional[Any]						  =    [dict(zip(A__   ,     A__						)						) for t in zip(*data.values()						)]
									a__  ,  a__ :Tuple						  =    self.scorer.predict(A__   ,     gpus=A__   ,     progress_bar=A__						)
									return {"mean_score": mean_score, "scores": scores}






from collections import OrderedDict
from typing import Mapping

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig


snake_case__			 =						{
    '''albert-base-v1''': '''https://huggingface.co/albert-base-v1/resolve/main/config.json''',
    '''albert-large-v1''': '''https://huggingface.co/albert-large-v1/resolve/main/config.json''',
    '''albert-xlarge-v1''': '''https://huggingface.co/albert-xlarge-v1/resolve/main/config.json''',
    '''albert-xxlarge-v1''': '''https://huggingface.co/albert-xxlarge-v1/resolve/main/config.json''',
    '''albert-base-v2''': '''https://huggingface.co/albert-base-v2/resolve/main/config.json''',
    '''albert-large-v2''': '''https://huggingface.co/albert-large-v2/resolve/main/config.json''',
    '''albert-xlarge-v2''': '''https://huggingface.co/albert-xlarge-v2/resolve/main/config.json''',
    '''albert-xxlarge-v2''': '''https://huggingface.co/albert-xxlarge-v2/resolve/main/config.json''',
}


class 					lowerCAmelCase_ (       _a):
						lowerCamelCase_         =     'albert'




						def __init__(	self						:   Any   ,     __A						:   Optional[int]=30000   ,     __A						:   List[str]=128   ,     __A						:   int=4096   ,     __A						:   Any=12   ,     __A						:   Union[str, Any]=1   ,     __A						:   Optional[int]=64   ,     __A						:   Dict=16384   ,     __A						:   List[str]=1   ,     __A						:   Any="gelu_new"   ,     __A						:   List[Any]=0   ,     __A						:   str=0   ,     __A						:   List[str]=512   ,     __A						:   Optional[Any]=2   ,     __A						:   Tuple=0.02   ,     __A						:   int=1E-12   ,     __A						:   str=0.1   ,     __A						:   Optional[Any]="absolute"   ,     __A						:   Tuple=0   ,     __A						:   str=2   ,     __A						:   Union[str, Any]=3   ,     **__A						:   Union[str, Any]   ,     )      ->str:






									"""simple docstring"""






									super().__init__(pad_token_id=__A   ,     bos_token_id=__A   ,     eos_token_id=__A   ,     **__A						)

									a__ :Tuple						  =    vocab_size
									a__ :Dict						  =    embedding_size
									a__ :Union[str, Any]						  =    hidden_size
									a__ :int						  =    num_hidden_layers
									a__ :List[Any]						  =    num_hidden_groups
									a__ :str						  =    num_attention_heads
									a__ :Optional[Any]						  =    inner_group_num
									a__ :Any						  =    hidden_act
									a__ :Optional[int]						  =    intermediate_size
									a__ :Optional[Any]						  =    hidden_dropout_prob
									a__ :Optional[int]						  =    attention_probs_dropout_prob
									a__ :int						  =    max_position_embeddings
									a__ :List[Any]						  =    type_vocab_size
									a__ :Dict						  =    initializer_range
									a__ :Optional[int]						  =    layer_norm_eps
									a__ :Optional[Any]						  =    classifier_dropout_prob
									a__ :Optional[Any]						  =    position_embedding_type






class 					lowerCAmelCase_ (       _a):




						@property
						def  _snake_case					(	self						:   Optional[int]						)      ->Mapping[str, Mapping[int, str]]:






									"""simple docstring"""






									if self.task == "multiple-choice":
												a__ :List[Any]						  =    {0: "batch", 1: "choice", 2: "sequence"}
									else:
												a__ :Dict						  =    {0: "batch", 1: "sequence"}
									return OrderedDict(
									    [
									        ("input_ids", dynamic_axis),
									        ("attention_mask", dynamic_axis),
									        ("token_type_ids", dynamic_axis),
									    ]						)


