




from dataclasses import dataclass, field
from typing import ClassVar, Dict

from ..features import Features, Value
from .base import TaskTemplate







@dataclass(frozen=lowerCamelCase_							)
class 				lowerCAmelCase_       (					lowerCamelCase_							):
				__a     :       str						=     field(default="language-modeling"						,     metadata={"include_in_asdict_even_if_is_default": True}							)

				__a     :       ClassVar[Features]						=     Features({"text": Value("string"							)}							)
				__a     :       ClassVar[Features]						=     Features({}							)
				__a     :       str						=     "text"






				@property
				def 		snake_case	(			self ):
							return {self.text_column: "text"}








'''simple docstring'''
import json
import os
from pathlib import Path
from shutil import copyfile
from typing import Any, Dict, List, Optional, Tuple, Union

import sentencepiece

from ...tokenization_utils import BatchEncoding, PreTrainedTokenizer
from ...utils import logging


snake_case_					:      Union[str, Any]    			=    logging.get_logger(__name__)

snake_case_					:      Tuple    			=    '▁'

snake_case_					:      List[str]    			=    {
    'vocab_file': 'vocab.json',
    'spm_file': 'sentencepiece.bpe.model',
    'tokenizer_config_file': 'tokenizer_config.json',
}

snake_case_					:      Tuple    			=    {
    'vocab_file': {
        'facebook/m2m100_418M': 'https://huggingface.co/facebook/m2m100_418M/resolve/main/vocab.json',
        'facebook/m2m100_1.2B': 'https://huggingface.co/facebook/m2m100_1.2B/resolve/main/vocab.json',
    },
    'spm_file': {
        'facebook/m2m100_418M': 'https://huggingface.co/facebook/m2m100_418M/resolve/main/sentencepiece.bpe.model',
        'facebook/m2m100_1.2B': 'https://huggingface.co/facebook/m2m100_1.2B/resolve/main/sentencepiece.bpe.model',
    },
    'tokenizer_config_file': {
        'facebook/m2m100_418M': 'https://huggingface.co/facebook/m2m100_418M/resolve/main/tokenizer_config.json',
        'facebook/m2m100_1.2B': 'https://huggingface.co/facebook/m2m100_1.2B/resolve/main/tokenizer_config.json',
    },
}

snake_case_					:      Union[str, Any]    			=    {
    'facebook/m2m100_418M': 1024,
}

# fmt: off
snake_case_					:      List[Any]    			=    {
    'm2m100': ['af', 'am', 'ar', 'ast', 'az', 'ba', 'be', 'bg', 'bn', 'br', 'bs', 'ca', 'ceb', 'cs', 'cy', 'da', 'de', 'el', 'en', 'es', 'et', 'fa', 'ff', 'fi', 'fr', 'fy', 'ga', 'gd', 'gl', 'gu', 'ha', 'he', 'hi', 'hr', 'ht', 'hu', 'hy', 'id', 'ig', 'ilo', 'is', 'it', 'ja', 'jv', 'ka', 'kk', 'km', 'kn', 'ko', 'lb', 'lg', 'ln', 'lo', 'lt', 'lv', 'mg', 'mk', 'ml', 'mn', 'mr', 'ms', 'my', 'ne', 'nl', 'no', 'ns', 'oc', 'or', 'pa', 'pl', 'ps', 'pt', 'ro', 'ru', 'sd', 'si', 'sk', 'sl', 'so', 'sq', 'sr', 'ss', 'su', 'sv', 'sw', 'ta', 'th', 'tl', 'tn', 'tr', 'uk', 'ur', 'uz', 'vi', 'wo', 'xh', 'yi', 'yo', 'zh', 'zu'],
    'wmt21': ['en', 'ha', 'is', 'ja', 'cs', 'ru', 'zh', 'de']
}






class 				lowercase__      (				lowercase  ):
	lowercase__    							=     VOCAB_FILES_NAMES
	lowercase__    							=     PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
	lowercase__    							=     PRETRAINED_VOCAB_FILES_MAP
	lowercase__    							=     ["""input_ids""", """attention_mask"""]

	lowercase__    							=     []
	lowercase__    							=     []





	def __init__(					self   :							Union[str, Any]					,lowerCamelCase__   :							List[str]					,lowerCamelCase__   :							List[str]					,lowerCamelCase__   :							Dict=None					,lowerCamelCase__   :							Optional[int]=None					,lowerCamelCase__   :							Optional[int]="<s>"					,lowerCamelCase__   :							Optional[int]="</s>"					,lowerCamelCase__   :							str="</s>"					,lowerCamelCase__   :							Union[str, Any]="<pad>"					,lowerCamelCase__   :							Optional[int]="<unk>"					,lowerCamelCase__   :							str="m2m100"					,lowerCamelCase__   :							Optional[Dict[str, Any]] = None					,lowerCamelCase__   :							Optional[Any]=8					,**lowerCamelCase__   :							str					,):







							'''simple docstring'''
							_UpperCamelCase   :						int												=  {} if sp_model_kwargs is None else sp_model_kwargs

							_UpperCamelCase   :						Optional[int]												=  language_codes
							_UpperCamelCase   :						Union[str, Any]												=  FAIRSEQ_LANGUAGE_CODES[language_codes]
							_UpperCamelCase   :						str												=  {lang_code: F'__{lang_code}__' for lang_code in fairseq_language_code}

							_UpperCamelCase   :						Optional[Any]												=  kwargs.get('additional_special_tokens'					,[]					)
							kwargs["additional_special_tokens"] += [
							    self.get_lang_token(lowerCamelCase__					)
							    for lang_code in fairseq_language_code
							    if self.get_lang_token(lowerCamelCase__					) not in kwargs["additional_special_tokens"]
							]

							super().__init__(
							    src_lang=lowerCamelCase__					,tgt_lang=lowerCamelCase__					,bos_token=lowerCamelCase__					,eos_token=lowerCamelCase__					,sep_token=lowerCamelCase__					,unk_token=lowerCamelCase__					,pad_token=lowerCamelCase__					,language_codes=lowerCamelCase__					,sp_model_kwargs=self.sp_model_kwargs					,num_madeup_words=lowerCamelCase__					,**lowerCamelCase__					,)

							_UpperCamelCase   :						Any												=  vocab_file
							_UpperCamelCase   :						int												=  load_json(lowerCamelCase__					)
							_UpperCamelCase   :						int												=  {v: k for k, v in self.encoder.items()}
							_UpperCamelCase   :						Any												=  spm_file
							_UpperCamelCase   :						Optional[Any]												=  load_spm(lowerCamelCase__					,self.sp_model_kwargs					)

							_UpperCamelCase   :						Union[str, Any]												=  len(self.encoder					)

							_UpperCamelCase   :						Dict												=  {
							    self.get_lang_token(lowerCamelCase__					): self.encoder_size + i for i, lang_code in enumerate(lowerCamelCase__					)
							}
							_UpperCamelCase   :						Any												=  {lang_code: self.encoder_size + i for i, lang_code in enumerate(lowerCamelCase__					)}
							_UpperCamelCase   :						Dict												=  {v: k for k, v in self.lang_token_to_id.items()}

							_UpperCamelCase   :						Any												=  src_lang if src_lang is not None else 'en'
							_UpperCamelCase   :						int												=  tgt_lang
							_UpperCamelCase   :						Optional[int]												=  self.get_lang_id(self._src_lang					)
							self.set_src_lang_special_tokens(self._src_lang					)

							_UpperCamelCase   :						List[Any]												=  num_madeup_words





	@property
	def       UpperCamelCase_    (					self   :							int					):







							'''simple docstring'''
							return len(self.encoder					) + len(self.lang_token_to_id					)





	@property
	def       UpperCamelCase_    (					self   :							Optional[int]					):







							'''simple docstring'''
							return self._src_lang





	@src_lang.setter
	def       UpperCamelCase_    (					self   :							List[Any]					,lowerCamelCase__   :							str					):







							'''simple docstring'''
							_UpperCamelCase   :						Tuple												=  new_src_lang
							self.set_src_lang_special_tokens(self._src_lang					)





	def       UpperCamelCase_    (					self   :							Union[str, Any]					,lowerCamelCase__   :							str					):







							'''simple docstring'''
							return self.sp_model.encode(lowerCamelCase__					,out_type=lowerCamelCase__					)





	def       UpperCamelCase_    (					self   :							Union[str, Any]					,lowerCamelCase__   :							str					):







							'''simple docstring'''
							if token in self.lang_token_to_id:
													return self.lang_token_to_id[token]
							return self.encoder.get(lowerCamelCase__					,self.encoder[self.unk_token]					)





	def       UpperCamelCase_    (					self   :							List[str]					,lowerCamelCase__   :							int					):







							'''simple docstring'''
							if index in self.id_to_lang_token:
													return self.id_to_lang_token[index]
							return self.decoder.get(lowerCamelCase__					,self.unk_token					)





	def       UpperCamelCase_    (					self   :							List[str]					,lowerCamelCase__   :							Union[str, Any]					):







							'''simple docstring'''
							_UpperCamelCase   :						List[Any]												=  []
							_UpperCamelCase   :						Union[str, Any]												=  ''
							for token in tokens:
													# make sure that special tokens are not decoded using sentencepiece model
													if token in self.all_special_tokens:
																			out_string += self.sp_model.decode(lowerCamelCase__					) + token
																			_UpperCamelCase   :						Optional[int]												=  []
													else:
																			current_sub_tokens.append(lowerCamelCase__					)
							out_string += self.sp_model.decode(lowerCamelCase__					)
							return out_string.strip()





	def       UpperCamelCase_    (					self   :							List[str]					,lowerCamelCase__   :							List[int]					,lowerCamelCase__   :							Optional[List[int]] = None					,lowerCamelCase__   :							bool = False					):







							'''simple docstring'''

							if already_has_special_tokens:
													return super().get_special_tokens_mask(
													    token_ids_a=lowerCamelCase__					,token_ids_a=lowerCamelCase__					,already_has_special_tokens=lowerCamelCase__					)

							_UpperCamelCase   :						Tuple												=  [1] * len(self.prefix_tokens					)
							_UpperCamelCase   :						str												=  [1] * len(self.suffix_tokens					)
							if token_ids_a is None:
													return prefix_ones + ([0] * len(lowerCamelCase__					)) + suffix_ones
							return prefix_ones + ([0] * len(lowerCamelCase__					)) + ([0] * len(lowerCamelCase__					)) + suffix_ones





	def       UpperCamelCase_    (					self   :							List[str]					,lowerCamelCase__   :							List[int]					,lowerCamelCase__   :							Optional[List[int]] = None					):







							'''simple docstring'''
							if token_ids_a is None:
													return self.prefix_tokens + token_ids_a + self.suffix_tokens
							# We don't expect to process pairs, but leave the pair logic for API consistency
							return self.prefix_tokens + token_ids_a + token_ids_a + self.suffix_tokens





	def       UpperCamelCase_    (					self   :							Dict					):







							'''simple docstring'''
							_UpperCamelCase   :						List[Any]												=  {self.convert_ids_to_tokens(lowerCamelCase__					): i for i in range(self.vocab_size					)}
							vocab.update(self.added_tokens_encoder					)
							return vocab





	def __getstate__(					self   :							Any					):







							'''simple docstring'''
							_UpperCamelCase   :						Dict												=  self.__dict__.copy()
							_UpperCamelCase   :						int												=  None
							return state





	def __setstate__(					self   :							Optional[int]					,lowerCamelCase__   :							Dict					):







							'''simple docstring'''
							_UpperCamelCase   :						Dict												=  d

							# for backward compatibility
							if not hasattr(self					,'sp_model_kwargs'					):
													_UpperCamelCase   :						int												=  {}

							_UpperCamelCase   :						Dict												=  load_spm(self.spm_file					,self.sp_model_kwargs					)





	def       UpperCamelCase_    (					self   :							str					,lowerCamelCase__   :							str					,lowerCamelCase__   :							Optional[str] = None					):







							'''simple docstring'''
							_UpperCamelCase   :						int												=  Path(lowerCamelCase__					)
							if not save_dir.is_dir():
													raise OSError(F'{save_directory} should be a directory'					)
							_UpperCamelCase   :						List[str]												=  save_dir / (
							    (filename_prefix + '-' if filename_prefix else '') + self.vocab_files_names['vocab_file']
							)
							_UpperCamelCase   :						Optional[Any]												=  save_dir / (
							    (filename_prefix + '-' if filename_prefix else '') + self.vocab_files_names['spm_file']
							)

							save_json(self.encoder					,lowerCamelCase__					)

							if os.path.abspath(self.spm_file					) != os.path.abspath(lowerCamelCase__					) and os.path.isfile(self.spm_file					):
													copyfile(self.spm_file					,lowerCamelCase__					)
							elif not os.path.isfile(self.spm_file					):
													with open(lowerCamelCase__					,'wb'					) as fi:
																			_UpperCamelCase   :						Union[str, Any]												=  self.sp_model.serialized_model_proto()
																			fi.write(lowerCamelCase__					)

							return (str(lowerCamelCase__					), str(lowerCamelCase__					))





	def       UpperCamelCase_    (					self   :							Tuple					,lowerCamelCase__   :							List[str]					,lowerCamelCase__   :							str = "en"					,lowerCamelCase__   :							Optional[List[str]] = None					,lowerCamelCase__   :							str = "ro"					,**lowerCamelCase__   :							List[Any]					,):







							'''simple docstring'''
							_UpperCamelCase   :						Optional[Any]												=  src_lang
							_UpperCamelCase   :						Any												=  tgt_lang
							self.set_src_lang_special_tokens(self.src_lang					)
							return super().prepare_seqaseq_batch(lowerCamelCase__					,lowerCamelCase__					,**lowerCamelCase__					)





	def       UpperCamelCase_    (					self   :							int					,lowerCamelCase__   :							List[str]					,lowerCamelCase__   :							Optional[str]					,lowerCamelCase__   :							Optional[str]					,**lowerCamelCase__   :							Optional[int]					):







							'''simple docstring'''
							if src_lang is None or tgt_lang is None:
													raise ValueError('Translation requires a `src_lang` and a `tgt_lang` for this model'					)
							_UpperCamelCase   :						int												=  src_lang
							_UpperCamelCase   :						Any												=  self(lowerCamelCase__					,add_special_tokens=lowerCamelCase__					,**lowerCamelCase__					)
							_UpperCamelCase   :						List[Any]												=  self.get_lang_id(lowerCamelCase__					)
							_UpperCamelCase   :						Any												=  tgt_lang_id
							return inputs





	def       UpperCamelCase_    (					self   :							Dict					):







							'''simple docstring'''
							self.set_src_lang_special_tokens(self.src_lang					)





	def       UpperCamelCase_    (					self   :							str					):







							'''simple docstring'''
							self.set_tgt_lang_special_tokens(self.tgt_lang					)





	def       UpperCamelCase_    (					self   :							str					,lowerCamelCase__   :							str					):







							'''simple docstring'''
							_UpperCamelCase   :						Union[str, Any]												=  self.get_lang_token(lowerCamelCase__					)
							_UpperCamelCase   :						str												=  self.lang_token_to_id[lang_token]
							_UpperCamelCase   :						Any												=  [self.cur_lang_id]
							_UpperCamelCase   :						Union[str, Any]												=  [self.eos_token_id]





	def       UpperCamelCase_    (					self   :							List[Any]					,lowerCamelCase__   :							str					):







							'''simple docstring'''
							_UpperCamelCase   :						Optional[Any]												=  self.get_lang_token(lowerCamelCase__					)
							_UpperCamelCase   :						Union[str, Any]												=  self.lang_token_to_id[lang_token]
							_UpperCamelCase   :						Union[str, Any]												=  [self.cur_lang_id]
							_UpperCamelCase   :						List[str]												=  [self.eos_token_id]





	def       UpperCamelCase_    (					self   :							Any					,lowerCamelCase__   :							str					):







							'''simple docstring'''
							return self.lang_code_to_token[lang]





	def       UpperCamelCase_    (					self   :							List[Any]					,lowerCamelCase__   :							str					):







							'''simple docstring'''
							_UpperCamelCase   :						List[str]												=  self.get_lang_token(lowerCamelCase__					)
							return self.lang_token_to_id[lang_token]






def A__       (	UpperCAmelCase_    ,    UpperCAmelCase_				):
						_UpperCamelCase   :						Union[str, Any]												=  sentencepiece.SentencePieceProcessor(**UpperCAmelCase_				)
						spm.Load(str(UpperCAmelCase_				)				)
						return spm






def A__       (	UpperCAmelCase_				):
						with open(UpperCAmelCase_    ,    'r'				) as f:
												return json.load(UpperCAmelCase_				)






def A__       (	UpperCAmelCase_    ,    UpperCAmelCase_				):
						with open(UpperCAmelCase_    ,    'w'				) as f:
												json.dump(UpperCAmelCase_    ,    UpperCAmelCase_    ,    indent=2				)


