



import json
import os
from typing import Optional, Tuple

import regex as re

from ...tokenization_utils import PreTrainedTokenizer
from ...utils import logging


snake_case    :						Any  =						logging.get_logger(__name__)

snake_case    :						List[str]  =						{
    'vocab_file': 'vocab.json',
    'merges_file': 'merges.txt',
}

snake_case    :						Optional[int]  =						{
    'vocab_file': {'ctrl': 'https://raw.githubusercontent.com/salesforce/ctrl/master/ctrl-vocab.json'},
    'merges_file': {'ctrl': 'https://raw.githubusercontent.com/salesforce/ctrl/master/ctrl-merges.txt'},
}

snake_case    :						List[Any]  =						{
    'ctrl': 2_5_6,
}

snake_case    :						List[str]  =						{
    'Pregnancy': 1_6_8_6_2_9,
    'Christianity': 7_6_7_5,
    'Explain': 1_0_6_4_2_3,
    'Fitness': 6_3_4_4_0,
    'Saving': 6_3_1_6_3,
    'Ask': 2_7_1_7_1,
    'Ass': 9_5_9_8_5,
    'Joke': 1_6_3_5_0_9,
    'Questions': 4_5_6_2_2,
    'Thoughts': 4_9_6_0_5,
    'Retail': 5_2_3_4_2,
    'Feminism': 1_6_4_3_3_8,
    'Writing': 1_1_9_9_2,
    'Atheism': 1_9_2_2_6_3,
    'Netflix': 4_8_6_1_6,
    'Computing': 3_9_6_3_9,
    'Opinion': 4_3_2_1_3,
    'Alone': 4_4_9_6_7,
    'Funny': 5_8_9_1_7,
    'Gaming': 4_0_3_5_8,
    'Human': 4_0_8_8,
    'India': 1_3_3_1,
    'Joker': 7_7_1_3_8,
    'Diet': 3_6_2_0_6,
    'Legal': 1_1_8_5_9,
    'Norman': 4_9_3_9,
    'Tip': 7_2_6_8_9,
    'Weight': 5_2_3_4_3,
    'Movies': 4_6_2_7_3,
    'Running': 2_3_4_2_5,
    'Science': 2_0_9_0,
    'Horror': 3_7_7_9_3,
    'Confession': 6_0_5_7_2,
    'Finance': 1_2_2_5_0,
    'Politics': 1_6_3_6_0,
    'Scary': 1_9_1_9_8_5,
    'Support': 1_2_6_5_4,
    'Technologies': 3_2_5_1_6,
    'Teenage': 6_6_1_6_0,
    'Event': 3_2_7_6_9,
    'Learned': 6_7_4_6_0,
    'Notion': 1_8_2_7_7_0,
    'Wikipedia': 3_7_5_8_3,
    'Books': 6_6_6_5,
    'Extract': 7_6_0_5_0,
    'Confessions': 1_0_2_7_0_1,
    'Conspiracy': 7_5_9_3_2,
    'Links': 6_3_6_7_4,
    'Narcissus': 1_5_0_4_2_5,
    'Relationship': 5_4_7_6_6,
    'Relationships': 1_3_4_7_9_6,
    'Reviews': 4_1_6_7_1,
    'News': 4_2_5_6,
    'Translation': 2_6_8_2_0,
    'multilingual': 1_2_8_4_0_6,
}
def 	snake_case__       (					__lowercase       )		->		Any:







						"""simple docstring"""

						A__	:							Dict						    = set()
						A__	:							Optional[int]						    = word[0]
						for char in word[1:]:
												pairs.add((prev_char, char)       )
												A__	:							str						    = char

						A__	:							str						    = set(__lowercase       )
						return pairs




class 							lowerCAmelCase__  (	UpperCamelCase						):
					__A     :    str           =						VOCAB_FILES_NAMES
					__A     :    Optional[int]           =						PRETRAINED_VOCAB_FILES_MAP
					__A     :    List[str]           =						PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
					__A     :    Union[str, Any]           =						CONTROL_CODES







					def __init__(       self      :     str			,		_A      :     Union[str, Any]			,		_A      :     Union[str, Any]			,		_A      :     Optional[int]="<unk>"			,		**_A      :     Dict):
											super().__init__(unk_token=_A			,		**_A)

											with open(_A			,		encoding="utf-8") as vocab_handle:
																	A__	:							List[str]						    = json.load(_A)
											A__	:							int						    = {v: k for k, v in self.encoder.items()}
											with open(_A			,		encoding="utf-8") as merges_handle:
																	A__	:							Optional[Any]						    = merges_handle.read().split("\n")[1:-1]
											A__	:							str						    = [tuple(merge.split()) for merge in merges]
											A__	:							List[Any]						    = dict(zip(_A			,		range(len(_A))))
											A__	:							int						    = {}







					@property
					def  _lowercase			(       self      :     Optional[int]):
											return len(self.encoder)







					def  _lowercase			(       self      :     Dict):
											return dict(self.encoder			,		**self.added_tokens_encoder)







					def  _lowercase			(       self      :     Dict			,		_A      :     Dict):
											if token in self.cache:
																	return self.cache[token]
											A__	:							Optional[int]						    = tuple(_A)
											A__	:							str						    = tuple(list(word[:-1]) + [word[-1] + "</w>"])
											A__	:							Dict						    = get_pairs(_A)

											if not pairs:
																	return token

											while True:
																	A__	:							List[str]						    = min(_A			,		key=lambda _A: self.bpe_ranks.get(_A			,		float("inf")))
																	if bigram not in self.bpe_ranks:
																							break
																	A__    ,       A__	:							List[str]						    = bigram
																	A__	:							Tuple						    = []
																	A__	:							Dict						    = 0
																	while i < len(_A):
																							try:
																													A__	:							Dict						    = word.index(_A			,		_A)
																							except ValueError:
																													new_word.extend(word[i:])
																													break
																							else:
																													new_word.extend(word[i:j])
																													A__	:							Tuple						    = j

																							if word[i] == first and i < len(_A) - 1 and word[i + 1] == second:
																													new_word.append(first + second)
																													i += 2
																							else:
																													new_word.append(word[i])
																													i += 1
																	A__	:							Union[str, Any]						    = tuple(_A)
																	A__	:							Optional[int]						    = new_word
																	if len(_A) == 1:
																							break
																	else:
																							A__	:							int						    = get_pairs(_A)
											A__	:							Dict						    = "@@ ".join(_A)
											A__	:							List[str]						    = word[:-4]
											A__	:							Optional[int]						    = word
											return word







					def  _lowercase			(       self      :     Optional[int]			,		_A      :     Dict):
											A__	:							Union[str, Any]						    = []

											A__	:							int						    = re.findall(r"\S+\n?"			,		_A)

											for token in words:
																	split_tokens.extend(list(self.bpe(_A).split(" ")))
											return split_tokens







					def  _lowercase			(       self      :     int			,		_A      :     Union[str, Any]):
											return self.encoder.get(_A			,		self.encoder.get(self.unk_token))







					def  _lowercase			(       self      :     List[Any]			,		_A      :     Optional[int]):
											return self.decoder.get(_A			,		self.unk_token)







					def  _lowercase			(       self      :     int			,		_A      :     Union[str, Any]):
											A__	:							str						    = " ".join(_A).replace("@@ "			,		"").strip()
											return out_string







					def  _lowercase			(       self      :     Any			,		_A      :     str			,		_A      :     Optional[str] = None):
											if not os.path.isdir(_A):
																	logger.error(F'Vocabulary path ({save_directory}) should be a directory')
																	return
											A__	:							List[str]						    = os.path.join(
											    _A			,		(filename_prefix + "-" if filename_prefix else "") + VOCAB_FILES_NAMES["vocab_file"])
											A__	:							str						    = os.path.join(
											    _A			,		(filename_prefix + "-" if filename_prefix else "") + VOCAB_FILES_NAMES["merges_file"])

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

											A__	:							Optional[int]						    = 0
											with open(_A			,		"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 _A: 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__	:							Optional[Any]						    = token_index
																							writer.write(" ".join(_A) + "\n")
																							index += 1

											return vocab_file, merge_file

					# def decode(self, token_ids, skip_special_tokens=False, clean_up_tokenization_spaces=True):
					#     filtered_tokens = ' '.join(self.convert_ids_to_tokens(token_ids, skip_special_tokens=skip_special_tokens))
					#     tokens_generated_so_far = re.sub('(@@ )', '', string=filtered_tokens)
					#     tokens_generated_so_far = re.sub('(@@ ?$)', '', string=tokens_generated_so_far)
					#     return ''.join(tokens_generated_so_far)




def 	snake_case__       (					__lowercase       )		->		int:







						"""simple docstring"""

						if divisor % 5 == 0 or divisor % 2 == 0:
												return 0
						A__	:							Tuple						    = 1
						A__	:							Union[str, Any]						    = 1
						while repunit:
												A__	:							Union[str, Any]						    = (1_0 * repunit + 1) % divisor
												repunit_index += 1
						return repunit_index
def 	snake_case__       (					__lowercase = 1_0_0_0_0_0_0       )		->		int:







						"""simple docstring"""

						A__	:							Dict						    = limit - 1
						if divisor % 2 == 0:
												divisor += 1
						while least_divisible_repunit(__lowercase       ) <= limit:
												divisor += 2
						return divisor


if __name__ == "__main__":
		print(f"""{solution() = }""")