





def      a    (   _lowercase :  list			,    _lowercase :  int			,    _lowercase :  int			,    _lowercase :  int							)  ->int:
		UpperCamelCase__								=						int((lower_bound + upper_bound) // 2							)
		if array[r] == value:
				return r
		if lower_bound >= upper_bound:
				return -1
		if array[r] < value:
				return binary_search(_lowercase			,    r + 1			,    _lowercase			,    _lowercase							)
		else:
				return binary_search(_lowercase			,    _lowercase			,    r - 1			,    _lowercase							)

def      a    (   _lowercase :  int			,    _lowercase :  list							)  ->list:
		UpperCamelCase__								=						0
		if matrix[index][0] == value:
				return [index, 0]
		while index < len(_lowercase							) and matrix[index][0] < value:
				UpperCamelCase__								=						binary_search(matrix[index]			,    0			,    len(matrix[index]							) - 1			,    _lowercase							)
				if r != -1:
						return [index, r]
				index += 1
		return [-1, -1]


if __name__ == "__main__":
				import doctest

				doctest.testmod()












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

import regex as re

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


__lowerCAmelCase 	=      logging.get_logger(__name__)


__lowerCAmelCase 	=      {"vocab_file": "vocab.json", "merges_file": "merges.txt"}

# See all MVP models at https://huggingface.co/models?filter=mvp
__lowerCAmelCase 	=      {
    "vocab_file": {
        "RUCAIBox/mvp": "https://huggingface.co/RUCAIBox/mvp/resolve/main/vocab.json",
    },
    "added_tokens.json": {
        "RUCAIBox/mvp": "https://huggingface.co/RUCAIBox/mvp/resolve/main/added_tokens.json",
    },
    "merges_file": {
        "RUCAIBox/mvp": "https://huggingface.co/RUCAIBox/mvp/resolve/main/merges.txt",
    },
}

__lowerCAmelCase 	=      {
    "RUCAIBox/mvp": 1_024,
}

@lru_cache()
def      a    (   )  ->Any:
		UpperCamelCase__								=						(
		    list(range(ord("""!"""							)			,    ord("""~"""							) + 1							)							) + list(range(ord("""¡"""							)			,    ord("""¬"""							) + 1							)							) + list(range(ord("""®"""							)			,    ord("""ÿ"""							) + 1							)							)
		)
		UpperCamelCase__								=						bs[:]
		UpperCamelCase__								=						0
		for b in range(2**8							):
				if b not in bs:
						bs.append(_lowercase							)
						cs.append(2**8 + n							)
						n += 1
		UpperCamelCase__								=						[chr(_lowercase							) for n in cs]
		return dict(zip(_lowercase			,    _lowercase							)							)

def      a    (   _lowercase :  int							)  ->List[Any]:
		UpperCamelCase__								=						set()
		UpperCamelCase__								=						word[0]
		for char in word[1:]:
				pairs.add((prev_char, char)							)
				UpperCamelCase__								=						char
		return pairs




class 	_UpperCAmelCase						( _lowercase       ):
					__A        = VOCAB_FILES_NAMES
					__A        = PRETRAINED_VOCAB_FILES_MAP
					__A        = PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
					__A        = ["input_ids", "attention_mask"]




					def __init__(    self     ,   __lowerCAmelCase     ,   __lowerCAmelCase     ,   __lowerCAmelCase="replace"     ,   __lowerCAmelCase="<s>"     ,   __lowerCAmelCase="</s>"     ,   __lowerCAmelCase="</s>"     ,   __lowerCAmelCase="<s>"     ,   __lowerCAmelCase="<unk>"     ,   __lowerCAmelCase="<pad>"     ,   __lowerCAmelCase="<mask>"     ,   __lowerCAmelCase=False     ,   **__lowerCAmelCase     ,   )    ->	str:
							UpperCamelCase__								=						AddedToken(__lowerCAmelCase     ,   lstrip=__lowerCAmelCase     ,   rstrip=__lowerCAmelCase							) if isinstance(__lowerCAmelCase     ,   __lowerCAmelCase							) else bos_token
							UpperCamelCase__								=						AddedToken(__lowerCAmelCase     ,   lstrip=__lowerCAmelCase     ,   rstrip=__lowerCAmelCase							) if isinstance(__lowerCAmelCase     ,   __lowerCAmelCase							) else eos_token
							UpperCamelCase__								=						AddedToken(__lowerCAmelCase     ,   lstrip=__lowerCAmelCase     ,   rstrip=__lowerCAmelCase							) if isinstance(__lowerCAmelCase     ,   __lowerCAmelCase							) else sep_token
							UpperCamelCase__								=						AddedToken(__lowerCAmelCase     ,   lstrip=__lowerCAmelCase     ,   rstrip=__lowerCAmelCase							) if isinstance(__lowerCAmelCase     ,   __lowerCAmelCase							) else cls_token
							UpperCamelCase__								=						AddedToken(__lowerCAmelCase     ,   lstrip=__lowerCAmelCase     ,   rstrip=__lowerCAmelCase							) if isinstance(__lowerCAmelCase     ,   __lowerCAmelCase							) else unk_token
							UpperCamelCase__								=						AddedToken(__lowerCAmelCase     ,   lstrip=__lowerCAmelCase     ,   rstrip=__lowerCAmelCase							) if isinstance(__lowerCAmelCase     ,   __lowerCAmelCase							) else pad_token

							# Mask token behave like a normal word, i.e. include the space before it
							UpperCamelCase__								=						AddedToken(__lowerCAmelCase     ,   lstrip=__lowerCAmelCase     ,   rstrip=__lowerCAmelCase							) if isinstance(__lowerCAmelCase     ,   __lowerCAmelCase							) else mask_token

							super().__init__(
							    errors=__lowerCAmelCase     ,   bos_token=__lowerCAmelCase     ,   eos_token=__lowerCAmelCase     ,   unk_token=__lowerCAmelCase     ,   sep_token=__lowerCAmelCase     ,   cls_token=__lowerCAmelCase     ,   pad_token=__lowerCAmelCase     ,   mask_token=__lowerCAmelCase     ,   add_prefix_space=__lowerCAmelCase     ,   **__lowerCAmelCase     ,   )
							with open(__lowerCAmelCase     ,   encoding="""utf-8"""							) as vocab_handle:
									UpperCamelCase__								=						json.load(__lowerCAmelCase							)
							UpperCamelCase__								=						{v: k for k, v in self.encoder.items()}
							UpperCamelCase__								=						errors  # how to handle errors in decoding
							UpperCamelCase__								=						bytes_to_unicode()
							UpperCamelCase__								=						{v: k for k, v in self.byte_encoder.items()}
							with open(__lowerCAmelCase     ,   encoding="""utf-8"""							) as merges_handle:
									UpperCamelCase__								=						merges_handle.read().split("""\n"""							)[1:-1]
							UpperCamelCase__								=						[tuple(merge.split()							) for merge in bpe_merges]
							UpperCamelCase__								=						dict(zip(__lowerCAmelCase     ,   range(len(__lowerCAmelCase							)							)							)							)
							UpperCamelCase__								=						{}
							UpperCamelCase__								=						add_prefix_space

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




					@property
					def 						snake_case_		(    self							)    ->	Union[str, Any]:
							return len(self.encoder							)




					def 						snake_case_		(    self							)    ->	Any:
							return dict(self.encoder     ,   **self.added_tokens_encoder							)




					def 						snake_case_		(    self     ,   __lowerCAmelCase							)    ->	List[str]:
							if token in self.cache:
									return self.cache[token]
							UpperCamelCase__								=						tuple(__lowerCAmelCase							)
							UpperCamelCase__								=						get_pairs(__lowerCAmelCase							)

							if not pairs:
									return token

							while True:
									UpperCamelCase__								=						min(__lowerCAmelCase     ,   key=lambda __lowerCAmelCase							: self.bpe_ranks.get(__lowerCAmelCase     ,   float("""inf"""							)							)							)
									if bigram not in self.bpe_ranks:
											break
									UpperCamelCase__	,      UpperCamelCase__								=						bigram
									UpperCamelCase__								=						[]
									UpperCamelCase__								=						0
									while i < len(__lowerCAmelCase							):
											try:
													UpperCamelCase__								=						word.index(__lowerCAmelCase     ,   __lowerCAmelCase							)
											except ValueError:
													new_word.extend(word[i:]							)
													break
											else:
													new_word.extend(word[i:j]							)
													UpperCamelCase__								=						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
									UpperCamelCase__								=						tuple(__lowerCAmelCase							)
									UpperCamelCase__								=						new_word
									if len(__lowerCAmelCase							) == 1:
											break
									else:
											UpperCamelCase__								=						get_pairs(__lowerCAmelCase							)
							UpperCamelCase__								=						""" """.join(__lowerCAmelCase							)
							UpperCamelCase__								=						word
							return word




					def 						snake_case_		(    self     ,   __lowerCAmelCase							)    ->	int:
							UpperCamelCase__								=						[]
							for token in re.findall(self.pat     ,   __lowerCAmelCase							):
									UpperCamelCase__								=						"""""".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 						snake_case_		(    self     ,   __lowerCAmelCase							)    ->	Dict:
							return self.encoder.get(__lowerCAmelCase     ,   self.encoder.get(self.unk_token							)							)




					def 						snake_case_		(    self     ,   __lowerCAmelCase							)    ->	Optional[Any]:
							return self.decoder.get(__lowerCAmelCase							)




					def 						snake_case_		(    self     ,   __lowerCAmelCase							)    ->	Any:
							UpperCamelCase__								=						"""""".join(__lowerCAmelCase							)
							UpperCamelCase__								=						bytearray([self.byte_decoder[c] for c in text]							).decode("""utf-8"""     ,   errors=self.errors							)
							return text




					def 						snake_case_		(    self     ,   __lowerCAmelCase     ,   __lowerCAmelCase = None							)    ->	Tuple[str]:
							if not os.path.isdir(__lowerCAmelCase							):
									logger.error(F"""Vocabulary path ({save_directory}) should be a directory"""							)
									return
							UpperCamelCase__								=						os.path.join(
							    __lowerCAmelCase     ,   (filename_prefix + """-""" if filename_prefix else """""") + VOCAB_FILES_NAMES["""vocab_file"""]							)
							UpperCamelCase__								=						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"""							)

							UpperCamelCase__								=						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!"""							)
													UpperCamelCase__								=						token_index
											writer.write(""" """.join(__lowerCAmelCase							) + """\n"""							)
											index += 1

							return vocab_file, merge_file




					def 						snake_case_		(    self     ,   __lowerCAmelCase     ,   __lowerCAmelCase = None							)    ->	List[int]:
							if token_ids_a is None:
									return [self.cls_token_id] + token_ids_a + [self.sep_token_id]
							UpperCamelCase__								=						[self.cls_token_id]
							UpperCamelCase__								=						[self.sep_token_id]
							return cls + token_ids_a + sep + sep + token_ids_a + sep




					def 						snake_case_		(    self     ,   __lowerCAmelCase     ,   __lowerCAmelCase = None     ,   __lowerCAmelCase = False							)    ->	List[int]:
							if already_has_special_tokens:
									return super().get_special_tokens_mask(
									    token_ids_a=__lowerCAmelCase     ,   token_ids_a=__lowerCAmelCase     ,   already_has_special_tokens=__lowerCAmelCase							)

							if token_ids_a is None:
									return [1] + ([0] * len(__lowerCAmelCase							)) + [1]
							return [1] + ([0] * len(__lowerCAmelCase							)) + [1, 1] + ([0] * len(__lowerCAmelCase							)) + [1]




					def 						snake_case_		(    self     ,   __lowerCAmelCase     ,   __lowerCAmelCase = None							)    ->	List[int]:
							UpperCamelCase__								=						[self.sep_token_id]
							UpperCamelCase__								=						[self.cls_token_id]

							if token_ids_a is None:
									return len(cls + token_ids_a + sep							) * [0]
							return len(cls + token_ids_a + sep + sep + token_ids_a + sep							) * [0]




					def 						snake_case_		(    self     ,   __lowerCAmelCase     ,   __lowerCAmelCase=False     ,   **__lowerCAmelCase							)    ->	Optional[Any]:
							UpperCamelCase__								=						kwargs.pop("""add_prefix_space"""     ,   self.add_prefix_space							)
							if (is_split_into_words or add_prefix_space) and (len(__lowerCAmelCase							) > 0 and not text[0].isspace()):
									UpperCamelCase__								=						""" """ + text
							return (text, kwargs)





