





'''simple docstring'''




# Usage:
# ./gen-card-facebook-wmt19.py

import os
from pathlib import Path
def 	UpperCAmelCase_							(					__lowercase							:      int			,     __lowercase							:      Optional[int]			,     __lowercase							:      Union[str, Any]       )		->       str:


     '''simple docstring'''



     _UpperCAmelCase     							=   {
         "en": "Machine learning is great, isn't it?",
         "ru": "Машинное обучение - это здорово, не так ли?",
         "de": "Maschinelles Lernen ist großartig, oder?",
     }

     # BLUE scores as follows:
     # "pair": [fairseq, transformers]
     _UpperCAmelCase     							=   {
         "ru-en": ["[41.3](http://matrix.statmt.org/matrix/output/1907?run_id=6937)", "39.20"],
         "en-ru": ["[36.4](http://matrix.statmt.org/matrix/output/1914?run_id=6724)", "33.47"],
         "en-de": ["[43.1](http://matrix.statmt.org/matrix/output/1909?run_id=6862)", "42.83"],
         "de-en": ["[42.3](http://matrix.statmt.org/matrix/output/1902?run_id=6750)", "41.35"],
     }
     _UpperCAmelCase     							=   f'{src_lang}-{tgt_lang}'

     _UpperCAmelCase     							=   f'\n---\nlanguage: \n- {src_lang}\n- {tgt_lang}\nthumbnail:\ntags:\n- translation\n- wmt19\n- facebook\nlicense: apache-2.0\ndatasets:\n- wmt19\nmetrics:\n- bleu\n---\n\n# FSMT\n\n## Model description\n\nThis is a ported version of [fairseq wmt19 transformer](https://github.com/pytorch/fairseq/blob/master/examples/wmt19/README.md) for {src_lang}-{tgt_lang}.\n\nFor more details, please see, [Facebook FAIR\'s WMT19 News Translation Task Submission](https://arxiv.org/abs/1907.06616).\n\nThe abbreviation FSMT stands for FairSeqMachineTranslation\n\nAll four models are available:\n\n* [wmt19-en-ru](https://huggingface.co/facebook/wmt19-en-ru)\n* [wmt19-ru-en](https://huggingface.co/facebook/wmt19-ru-en)\n* [wmt19-en-de](https://huggingface.co/facebook/wmt19-en-de)\n* [wmt19-de-en](https://huggingface.co/facebook/wmt19-de-en)\n\n## Intended uses & limitations\n\n#### How to use\n\n```python\nfrom transformers import FSMTForConditionalGeneration, FSMTTokenizer\nmname = "facebook/wmt19-{src_lang}-{tgt_lang}"\ntokenizer = FSMTTokenizer.from_pretrained(mname)\nmodel = FSMTForConditionalGeneration.from_pretrained(mname)\n\ninput = "{texts[src_lang]}"\ninput_ids = tokenizer.encode(input, return_tensors="pt")\noutputs = model.generate(input_ids)\ndecoded = tokenizer.decode(outputs[0], skip_special_tokens=True)\nprint(decoded) # {texts[tgt_lang]}\n\n```\n\n#### Limitations and bias\n\n- The original (and this ported model) doesn\'t seem to handle well inputs with repeated sub-phrases, [content gets truncated](https://discuss.huggingface.co/t/issues-with-translating-inputs-containing-repeated-phrases/981)\n\n## Training data\n\nPretrained weights were left identical to the original model released by fairseq. For more details, please, see the [paper](https://arxiv.org/abs/1907.06616).\n\n## Eval results\n\npair   | fairseq | transformers\n-------|---------|----------\n{pair}  | {scores[pair][0]} | {scores[pair][1]}\n\nThe score is slightly below the score reported by `fairseq`, since `transformers`` currently doesn\'t support:\n- model ensemble, therefore the best performing checkpoint was ported (``model4.pt``).\n- re-ranking\n\nThe score was calculated using this code:\n\n```bash\ngit clone https://github.com/huggingface/transformers\ncd transformers\nexport PAIR={pair}\nexport DATA_DIR=data/$PAIR\nexport SAVE_DIR=data/$PAIR\nexport BS=8\nexport NUM_BEAMS=15\nmkdir -p $DATA_DIR\nsacrebleu -t wmt19 -l $PAIR --echo src > $DATA_DIR/val.source\nsacrebleu -t wmt19 -l $PAIR --echo ref > $DATA_DIR/val.target\necho $PAIR\nPYTHONPATH="src:examples/seq2seq" python examples/seq2seq/run_eval.py facebook/wmt19-$PAIR $DATA_DIR/val.source $SAVE_DIR/test_translations.txt --reference_path $DATA_DIR/val.target --score_path $SAVE_DIR/test_bleu.json --bs $BS --task translation --num_beams $NUM_BEAMS\n```\nnote: fairseq reports using a beam of 50, so you should get a slightly higher score if re-run with `--num_beams 50`.\n\n## Data Sources\n\n- [training, etc.](http://www.statmt.org/wmt19/)\n- [test set](http://matrix.statmt.org/test_sets/newstest2019.tgz?1556572561)\n\n\n### BibTeX entry and citation info\n\n```bibtex\n@inproceedings{{...,\n  year={{2020}},\n  title={{Facebook FAIR\'s WMT19 News Translation Task Submission}},\n  author={{Ng, Nathan and Yee, Kyra and Baevski, Alexei and Ott, Myle and Auli, Michael and Edunov, Sergey}},\n  booktitle={{Proc. of WMT}},\n}}\n```\n\n\n## TODO\n\n- port model ensemble (fairseq uses 4 model checkpoints)\n\n'
     os.makedirs(__lowercase			,     exist_ok=__lowercase       )
     _UpperCAmelCase     							=   os.path.join(__lowercase			,     "README.md"       )
     print(f'Generating {path}'       )
     with open(__lowercase			,     "w"			,     encoding="utf-8"       ) as f:
          f.write(__lowercase       )


# make sure we are under the root of the project
__SCREAMING_SNAKE_CASE     :Optional[Any]       =				Path(__file__).resolve().parent.parent.parent
__SCREAMING_SNAKE_CASE     :Any       =				repo_dir / '''model_cards'''

for model_name in ["wmt19-ru-en", "wmt19-en-ru", "wmt19-en-de", "wmt19-de-en"]:
  __SCREAMING_SNAKE_CASE  ,__SCREAMING_SNAKE_CASE  ,__SCREAMING_SNAKE_CASE     :Optional[Any]       =				model_name.split('''-''')
  __SCREAMING_SNAKE_CASE     :List[str]       =				model_cards_dir / '''facebook''' / model_name
  write_model_card(model_card_dir, src_lang=src_lang, tgt_lang=tgt_lang)





'''simple docstring'''




import copy
from typing import Any, Dict, List, Optional, Union

import numpy as np

from ...audio_utils import mel_filter_bank, spectrogram, window_function
from ...feature_extraction_sequence_utils import SequenceFeatureExtractor
from ...feature_extraction_utils import BatchFeature
from ...utils import TensorType, logging


__lowercase          =  logging.get_logger(__name__)






class 				a__(					lowerCAmelCase__	):


			'''simple docstring'''






			UpperCAmelCase_			:     Union[str, Any]      =       ['''input_features''']



			def __init__(   self       ,    __lowerCAmelCase=80       ,    __lowerCAmelCase=16000       ,    __lowerCAmelCase=160       ,    __lowerCAmelCase=30       ,    __lowerCAmelCase=400       ,    __lowerCAmelCase=0.0       ,    __lowerCAmelCase=False       ,    **__lowerCAmelCase       ,    ):




					"""simple docstring"""


					super().__init__(
					    feature_size=__lowerCAmelCase       ,    sampling_rate=__lowerCAmelCase       ,    padding_value=__lowerCAmelCase       ,    return_attention_mask=__lowerCAmelCase       ,    **__lowerCAmelCase       ,    )
					lowerCAmelCase   					=  n_fft
					lowerCAmelCase   					=  hop_length
					lowerCAmelCase   					=  chunk_length
					lowerCAmelCase   					=  chunk_length * sampling_rate
					lowerCAmelCase   					=  self.n_samples // hop_length
					lowerCAmelCase   					=  sampling_rate
					lowerCAmelCase   					=  mel_filter_bank(
					    num_frequency_bins=1 + n_fft // 2       ,    num_mel_filters=__lowerCAmelCase       ,    min_frequency=0.0       ,    max_frequency=8000.0       ,    sampling_rate=__lowerCAmelCase       ,    norm="""slaney"""       ,    mel_scale="""slaney"""       ,    )



			def 							a_       (   self       ,    __lowerCAmelCase):




					"""simple docstring"""


					lowerCAmelCase   					=  spectrogram(
					    __lowerCAmelCase       ,    window_function(self.n_fft       ,    """hann""")       ,    frame_length=self.n_fft       ,    hop_length=self.hop_length       ,    power=2.0       ,    mel_filters=self.mel_filters       ,    log_mel="""log10"""       ,    )
					lowerCAmelCase   					=  log_spec[:, :-1]
					lowerCAmelCase   					=  np.maximum(__lowerCAmelCase       ,    log_spec.max() - 8.0)
					lowerCAmelCase   					=  (log_spec + 4.0) / 4.0
					return log_spec



			@staticmethod
			# Copied from transformers.models.wav2vec2.feature_extraction_wav2vec2.Wav2Vec2FeatureExtractor.zero_mean_unit_var_norm
			def 							a_       (   __lowerCAmelCase       ,    __lowerCAmelCase       ,    __lowerCAmelCase = 0.0):




					"""simple docstring"""


					if attention_mask is not None:
							lowerCAmelCase   					=  np.array(__lowerCAmelCase       ,    np.intaa)
							lowerCAmelCase   					=  []

							for vector, length in zip(__lowerCAmelCase       ,    attention_mask.sum(-1)):
									lowerCAmelCase   					=  (vector - vector[:length].mean()) / np.sqrt(vector[:length].var() + 1E-7)
									if length < normed_slice.shape[0]:
											lowerCAmelCase   					=  padding_value

									normed_input_values.append(__lowerCAmelCase)
					else:
							lowerCAmelCase   					=  [(x - x.mean()) / np.sqrt(x.var() + 1E-7) for x in input_values]

					return normed_input_values



			def __call__(   self       ,    __lowerCAmelCase       ,    __lowerCAmelCase = True       ,    __lowerCAmelCase = None       ,    __lowerCAmelCase = None       ,    __lowerCAmelCase = None       ,    __lowerCAmelCase = "max_length"       ,    __lowerCAmelCase = None       ,    __lowerCAmelCase = None       ,    __lowerCAmelCase = None       ,    **__lowerCAmelCase       ,    ):




					"""simple docstring"""


					if sampling_rate is not None:
							if sampling_rate != self.sampling_rate:
									raise ValueError(
									    f"The model corresponding to this feature extractor: {self.__class__.__name__} was trained using a"
									    f" sampling rate of {self.sampling_rate}. Please make sure that the provided `raw_speech` input"
									    f" was sampled with {self.sampling_rate} and not {sampling_rate}.")
					else:
							logger.warning(
							    """It is strongly recommended to pass the `sampling_rate` argument to this function. """
							    """Failing to do so can result in silent errors that might be hard to debug.""")

					lowerCAmelCase   					=  isinstance(__lowerCAmelCase       ,    np.ndarray) and len(raw_speech.shape) > 1
					if is_batched_numpy and len(raw_speech.shape) > 2:
							raise ValueError(f"Only mono-channel audio is supported for input to {self}")
					lowerCAmelCase   					=  is_batched_numpy or (
					    isinstance(__lowerCAmelCase       ,    (list, tuple)) and (isinstance(raw_speech[0]       ,    (np.ndarray, tuple, list)))
					)

					if is_batched:
							lowerCAmelCase   					=  [np.asarray([speech]       ,    dtype=np.floataa).T for speech in raw_speech]
					elif not is_batched and not isinstance(__lowerCAmelCase       ,    np.ndarray):
							lowerCAmelCase   					=  np.asarray(__lowerCAmelCase       ,    dtype=np.floataa)
					elif isinstance(__lowerCAmelCase       ,    np.ndarray) and raw_speech.dtype is np.dtype(np.floataa):
							lowerCAmelCase   					=  raw_speech.astype(np.floataa)

					# always return batch
					if not is_batched:
							lowerCAmelCase   					=  [np.asarray([raw_speech]).T]

					lowerCAmelCase   					=  BatchFeature({"""input_features""": raw_speech})

					# convert into correct format for padding

					lowerCAmelCase   					=  self.pad(
					    __lowerCAmelCase       ,    padding=__lowerCAmelCase       ,    max_length=max_length if max_length else self.n_samples       ,    truncation=__lowerCAmelCase       ,    pad_to_multiple_of=__lowerCAmelCase       ,    return_attention_mask=return_attention_mask or do_normalize       ,    )

					# zero-mean and unit-variance normalization
					if do_normalize:
							lowerCAmelCase   					=  self.zero_mean_unit_var_norm(
							    padded_inputs["""input_features"""]       ,    attention_mask=padded_inputs["""attention_mask"""]       ,    padding_value=self.padding_value       ,    )
							lowerCAmelCase   					=  np.stack(padded_inputs["""input_features"""]       ,    axis=0)

					# make sure list is in array format
					lowerCAmelCase   					=  padded_inputs.get("""input_features""").transpose(2       ,    0       ,    1)

					lowerCAmelCase   					=  [self._np_extract_fbank_features(__lowerCAmelCase) for waveform in input_features[0]]

					if isinstance(input_features[0]       ,    __lowerCAmelCase):
							lowerCAmelCase   					=  [np.asarray(__lowerCAmelCase       ,    dtype=np.floataa) for feature in input_features]
					else:
							lowerCAmelCase   					=  input_features

					if return_attention_mask:
							# rescale from sample (48000) to feature (3000)
							lowerCAmelCase   					=  padded_inputs["""attention_mask"""][:, :: self.hop_length]

					if return_tensors is not None:
							lowerCAmelCase   					=  padded_inputs.convert_to_tensors(__lowerCAmelCase)

					return padded_inputs







			def 							a_       (   self):




					"""simple docstring"""


					lowerCAmelCase   					=  copy.deepcopy(self.__dict__)
					lowerCAmelCase   					=  self.__class__.__name__
					if "mel_filters" in output:
							del output["mel_filters"]
					return output




