




'''simple docstring'''


import json
import os
from functools import lru_cache
from typing import TYPE_CHECKING, List, Optional, Tuple, Union

import numpy as np
import regex as re

from ...utils import is_tf_available, is_torch_available, logging


if TYPE_CHECKING:
				if is_torch_available():
								import torch
				if is_tf_available():
								import tensorflow as tf

from ...tokenization_utils import AddedToken, PreTrainedTokenizer


_snake_case      :      str   							=		logging.get_logger(__name__)

_snake_case      :      str   							=		{
    'vocab_file': 'vocab.json',
    'merges_file': 'merges.txt',
}

_snake_case      :      Optional[Any]   							=		{
    'vocab_file': {
        'Salesforce/codegen-350M-mono': 'https://huggingface.co/Salesforce/codegen-350M-mono/resolve/main/vocab.json',
    },
    'merges_file': {
        'Salesforce/codegen-350M-mono': 'https://huggingface.co/Salesforce/codegen-350M-mono/resolve/main/merges.txt',
    },
}

_snake_case      :      Optional[int]   							=		{
    'Salesforce/codegen-350M-mono': 2048,
}
@lru_cache()
def 		snake_case_    ():





							'''simple docstring'''


							_a	     =				(
							    list(range(ord('''!'''      ) ,						ord('''~'''      ) + 1      )      ) + list(range(ord('''¡'''      ) ,						ord('''¬'''      ) + 1      )      ) + list(range(ord('''®'''      ) ,						ord('''ÿ'''      ) + 1      )      )
							)
							_a	     =				bs[:]
							_a	     =				0
							for b in range(2**8      ):
														if b not in bs:
																					bs.append(UpperCamelCase      )
																					cs.append(2**8 + n      )
																					n += 1
							_a	     =				[chr(UpperCamelCase      ) for n in cs]
							return dict(zip(UpperCamelCase ,						UpperCamelCase      )      )
def 		snake_case_    (UpperCamelCase       :  Tuple      ):





							'''simple docstring'''


							_a	     =				set()
							_a	     =				word[0]
							for char in word[1:]:
														pairs.add((prev_char, char)      )
														_a	     =				char
							return pairs

class 			A    (		_a    ):
							lowercase_							=				VOCAB_FILES_NAMES
							lowercase_							=				PRETRAINED_VOCAB_FILES_MAP
							lowercase_							=				PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
							lowercase_							=				['input_ids', 'attention_mask']







							def __init__(  self       :  Optional[Any]							,			lowerCAmelCase_       :  List[str]							,			lowerCAmelCase_       :  List[Any]							,			lowerCAmelCase_       :  Tuple="replace"							,			lowerCAmelCase_       :  List[Any]="<|endoftext|>"							,			lowerCAmelCase_       :  Tuple="<|endoftext|>"							,			lowerCAmelCase_       :  List[str]="<|endoftext|>"							,			lowerCAmelCase_       :  str=None							,			lowerCAmelCase_       :  Tuple=False							,			lowerCAmelCase_       :  Tuple=False							,			**lowerCAmelCase_       :  Optional[Any]							,			)			->							List[Any]:


														"""simple docstring"""


														_a	     =				AddedToken(lowerCAmelCase_							,			lstrip=lowerCAmelCase_							,			rstrip=lowerCAmelCase_					) if isinstance(lowerCAmelCase_							,			lowerCAmelCase_					) else bos_token
														_a	     =				AddedToken(lowerCAmelCase_							,			lstrip=lowerCAmelCase_							,			rstrip=lowerCAmelCase_					) if isinstance(lowerCAmelCase_							,			lowerCAmelCase_					) else eos_token
														_a	     =				AddedToken(lowerCAmelCase_							,			lstrip=lowerCAmelCase_							,			rstrip=lowerCAmelCase_					) if isinstance(lowerCAmelCase_							,			lowerCAmelCase_					) else unk_token
														_a	     =				AddedToken(lowerCAmelCase_							,			lstrip=lowerCAmelCase_							,			rstrip=lowerCAmelCase_					) if isinstance(lowerCAmelCase_							,			lowerCAmelCase_					) else pad_token
														super().__init__(
														    errors=lowerCAmelCase_							,			unk_token=lowerCAmelCase_							,			bos_token=lowerCAmelCase_							,			eos_token=lowerCAmelCase_							,			pad_token=lowerCAmelCase_							,			add_prefix_space=lowerCAmelCase_							,			add_bos_token=lowerCAmelCase_							,			**lowerCAmelCase_							,			)
														_a	     =				add_bos_token

														with open(lowerCAmelCase_							,			encoding='''utf-8'''					) as vocab_handle:
																					_a	     =				json.load(lowerCAmelCase_					)
														_a	     =				{v: k for k, v in self.encoder.items()}
														_a	     =				errors  # how to handle errors in decoding
														_a	     =				bytes_to_unicode()
														_a	     =				{v: k for k, v in self.byte_encoder.items()}
														with open(lowerCAmelCase_							,			encoding='''utf-8'''					) as merges_handle:
																					_a	     =				merges_handle.read().split('''\n'''					)[1:-1]
														_a	     =				[tuple(merge.split()					) for merge in bpe_merges]
														_a	     =				dict(zip(lowerCAmelCase_							,			range(len(lowerCAmelCase_					)					)					)					)
														_a	     =				{}
														_a	     =				add_prefix_space

														# Should have added re.IGNORECASE so BPE merges can happen for capitalized versions of contractions
														_a	     =				re.compile(R'''\'s|\'t|\'re|\'ve|\'m|\'ll|\'d| ?\p{L}+| ?\p{N}+| ?[^\s\p{L}\p{N}]+|\s+(?!\S)|\s+'''					)







							@property
							def    __lowerCAmelCase			(  self       :  Union[str, Any]					)			->							Optional[Any]:


														"""simple docstring"""


														return len(self.encoder					)







							def    __lowerCAmelCase			(  self       :  Union[str, Any]					)			->							str:


														"""simple docstring"""


														return dict(self.encoder							,			**self.added_tokens_encoder					)







							def    __lowerCAmelCase			(  self       :  Optional[int]							,			lowerCAmelCase_       :  Union[str, Any]					)			->							Tuple:


														"""simple docstring"""


														if token in self.cache:
																					return self.cache[token]
														_a	     =				tuple(lowerCAmelCase_					)
														_a	     =				get_pairs(lowerCAmelCase_					)

														if not pairs:
																					return token

														while True:
																					_a	     =				min(lowerCAmelCase_							,			key=lambda lowerCAmelCase_					: self.bpe_ranks.get(lowerCAmelCase_							,			float('''inf'''					)					)					)
																					if bigram not in self.bpe_ranks:
																												break
																					_a     ,							_a	     =				bigram
																					_a	     =				[]
																					_a	     =				0
																					while i < len(lowerCAmelCase_					):
																												try:
																																			_a	     =				word.index(lowerCAmelCase_							,			lowerCAmelCase_					)
																												except ValueError:
																																			new_word.extend(word[i:]					)
																																			break
																												else:
																																			new_word.extend(word[i:j]					)
																																			_a	     =				j

																												if word[i] == first and i < len(lowerCAmelCase_					) - 1 and word[i + 1] == second:
																																			new_word.append(first + second					)
																																			i += 2
																												else:
																																			new_word.append(word[i]					)
																																			i += 1
																					_a	     =				tuple(lowerCAmelCase_					)
																					_a	     =				new_word
																					if len(lowerCAmelCase_					) == 1:
																												break
																					else:
																												_a	     =				get_pairs(lowerCAmelCase_					)
														_a	     =				''' '''.join(lowerCAmelCase_					)
														_a	     =				word
														return word







							def    __lowerCAmelCase			(  self       :  List[str]							,			lowerCAmelCase_       :  Any							,			lowerCAmelCase_       :  Union[str, Any]=None					)			->							List[Any]:


														"""simple docstring"""


														if self.add_bos_token:
																					_a	     =				[self.bos_token_id]
														else:
																					_a	     =				[]

														_a	     =				bos_token_ids + token_ids_a

														if token_ids_a is None:
																					return output

														return output + bos_token_ids + token_ids_a







							def    __lowerCAmelCase			(  self       :  Optional[int]							,			lowerCAmelCase_       :  Optional[Any]					)			->							Optional[int]:


														"""simple docstring"""


														_a	     =				[]
														for token in re.findall(self.pat							,			lowerCAmelCase_					):
																					_a	     =				''''''.join(
																					    self.byte_encoder[b] for b in token.encode('''utf-8'''					)					)  # Maps all our bytes to unicode strings, avoiding control tokens of the BPE (spaces in our case)
																					bpe_tokens.extend(bpe_token for bpe_token in self.bpe(lowerCAmelCase_					).split(''' '''					)					)
														return bpe_tokens







							def    __lowerCAmelCase			(  self       :  Tuple							,			lowerCAmelCase_       :  Tuple					)			->							Optional[int]:


														"""simple docstring"""


														return self.encoder.get(lowerCAmelCase_							,			self.encoder.get(self.unk_token					)					)







							def    __lowerCAmelCase			(  self       :  List[str]							,			lowerCAmelCase_       :  Tuple					)			->							Union[str, Any]:


														"""simple docstring"""


														return self.decoder.get(lowerCAmelCase_					)







							def    __lowerCAmelCase			(  self       :  Union[str, Any]							,			lowerCAmelCase_       :  Optional[int]					)			->							List[str]:


														"""simple docstring"""


														_a	     =				''''''.join(lowerCAmelCase_					)
														_a	     =				bytearray([self.byte_decoder[c] for c in text]					).decode('''utf-8'''							,			errors=self.errors					)
														return text







							def    __lowerCAmelCase			(  self       :  str							,			lowerCAmelCase_       :  str							,			lowerCAmelCase_       :  Optional[str] = None					)			->							Tuple[str]:


														"""simple docstring"""


														if not os.path.isdir(lowerCAmelCase_					):
																					logger.error(F'Vocabulary path ({save_directory}) should be a directory'					)
																					return
														_a	     =				os.path.join(
														    lowerCAmelCase_							,			(filename_prefix + '''-''' if filename_prefix else '''''') + VOCAB_FILES_NAMES['''vocab_file''']					)
														_a	     =				os.path.join(
														    lowerCAmelCase_							,			(filename_prefix + '''-''' if filename_prefix else '''''') + VOCAB_FILES_NAMES['''merges_file''']					)

														with open(lowerCAmelCase_							,			'''w'''							,			encoding='''utf-8'''					) as f:
																					f.write(json.dumps(self.encoder							,			indent=2							,			sort_keys=lowerCAmelCase_							,			ensure_ascii=lowerCAmelCase_					) + '''\n'''					)

														_a	     =				0
														with open(lowerCAmelCase_							,			'''w'''							,			encoding='''utf-8'''					) as writer:
																					writer.write('''#version: 0.2\n'''					)
																					for bpe_tokens, token_index in sorted(self.bpe_ranks.items()							,			key=lambda lowerCAmelCase_					: kv[1]					):
																												if index != token_index:
																																			logger.warning(
																																			    F'Saving vocabulary to {merge_file}: BPE merge indices are not consecutive.'
																																			    ''' Please check that the tokenizer is not corrupted!'''					)
																																			_a	     =				token_index
																												writer.write(''' '''.join(lowerCAmelCase_					) + '''\n'''					)
																												index += 1

														return vocab_file, merge_file







							def    __lowerCAmelCase			(  self       :  str							,			lowerCAmelCase_       :  Any							,			lowerCAmelCase_       :  List[Any]=False							,			**lowerCAmelCase_       :  List[Any]					)			->							List[str]:


														"""simple docstring"""


														_a	     =				kwargs.pop('''add_prefix_space'''							,			self.add_prefix_space					)
														if is_split_into_words or add_prefix_space:
																					_a	     =				''' ''' + text
														return (text, kwargs)







							def    __lowerCAmelCase			(  self       :  Optional[Any]							,			lowerCAmelCase_       :  Union[int, List[int], "np.ndarray", "torch.Tensor", "tf.Tensor"]							,			lowerCAmelCase_       :  bool = False							,			lowerCAmelCase_       :  bool = None							,			lowerCAmelCase_       :  Optional[List[str]] = None							,			**lowerCAmelCase_       :  List[str]							,			)			->							str:


														"""simple docstring"""


														_a	     =				super()._decode(
														    token_ids=lowerCAmelCase_							,			skip_special_tokens=lowerCAmelCase_							,			clean_up_tokenization_spaces=lowerCAmelCase_							,			**lowerCAmelCase_							,			)

														if truncate_before_pattern is not None and len(lowerCAmelCase_					) > 0:
																					_a	     =				self.truncate(lowerCAmelCase_							,			lowerCAmelCase_					)

														return decoded_text







							def    __lowerCAmelCase			(  self       :  List[Any]							,			lowerCAmelCase_       :  Union[str, Any]							,			lowerCAmelCase_       :  Optional[Any]					)			->							Optional[int]:


														"""simple docstring"""


														def find_re(lowerCAmelCase_       :  str							,			lowerCAmelCase_       :  Optional[Any]							,			lowerCAmelCase_       :  Optional[int]					):
																					_a	     =				pattern.search(lowerCAmelCase_							,			lowerCAmelCase_					)
																					return m.start() if m else -1

														_a	     =				[re.compile(lowerCAmelCase_							,			re.MULTILINE					) for pattern in truncate_before_pattern]

														_a	     =				list(re.finditer('''^print'''							,			lowerCAmelCase_							,			re.MULTILINE					)					)

														if len(lowerCAmelCase_					) > 1:
																					_a	     =				completion[: prints[1].start()]

														_a	     =				list(re.finditer('''^def'''							,			lowerCAmelCase_							,			re.MULTILINE					)					)

														if len(lowerCAmelCase_					) > 1:
																					_a	     =				completion[: defs[1].start()]

														_a	     =				0

														_a	     =				[
														    pos for pos in [find_re(lowerCAmelCase_							,			lowerCAmelCase_							,			lowerCAmelCase_					) for terminal in terminals] if pos != -1
														]

														if len(lowerCAmelCase_					) > 0:
																					return completion[: min(lowerCAmelCase_					)]
														else:
																					return completion







'''simple docstring'''


from dataclasses import dataclass
from typing import Callable, Optional

import datasets

@dataclass
class 			A    (		datasets.BuilderConfig    ):
							lowercase_							=				None
							lowercase_							=				None
							lowercase_							=				None







							def    __lowerCAmelCase			(  self       :  Dict					)			->							List[str]:


														"""simple docstring"""


														assert self.generator is not None, "generator must be specified"

														if self.gen_kwargs is None:
																					_a	     =				{}



class 			A    (		datasets.GeneratorBasedBuilder    ):
							lowercase_							=				GeneratorConfig







							def    __lowerCAmelCase			(  self       :  Any					)			->							List[str]:


														"""simple docstring"""


														return datasets.DatasetInfo(features=self.config.features					)







							def    __lowerCAmelCase			(  self       :  Dict							,			lowerCAmelCase_       :  Optional[int]					)			->							Optional[Any]:


														"""simple docstring"""


														return [datasets.SplitGenerator(name=datasets.Split.TRAIN							,			gen_kwargs=self.config.gen_kwargs					)]







							def    __lowerCAmelCase			(  self       :  str							,			**lowerCAmelCase_       :  int					)			->							List[str]:


														"""simple docstring"""


														for idx, ex in enumerate(self.config.generator(**lowerCAmelCase_					)					):
																					yield idx, ex

