






'''simple docstring'''



import math
import unittest



def  __UpperCAmelCase						(					SCREAMING_SNAKE_CASE__:						int     )	->      bool:

   """simple docstring"""
   assert isinstance(SCREAMING_SNAKE_CASE__,       SCREAMING_SNAKE_CASE__     ) and (
       number >= 0
   ), "'number' must been an int and positive"

   if 1 < number < 4:
      # 2 and 3 are primes
      return True
   elif number < 2 or number % 2 == 0 or number % 3 == 0:
      # Negatives, 0, 1, all even numbers, all multiples of 3 are not primes
      return False

   # All primes number are in format of 6k +/- 1
   for i in range(5,       int(math.sqrt(SCREAMING_SNAKE_CASE__     ) + 1     ),       6     ):
      if number % i == 0 or number % (i + 2) == 0:
         return False
   return True
class 				__SCREAMING_SNAKE_CASE							(				unittest.TestCase  ):




      def 			__UpperCamelCase				(	self   )				->str:



         '''simple docstring'''
         self.assertTrue(is_prime(2   )   )
         self.assertTrue(is_prime(3   )   )
         self.assertTrue(is_prime(5   )   )
         self.assertTrue(is_prime(7   )   )
         self.assertTrue(is_prime(11   )   )
         self.assertTrue(is_prime(13   )   )
         self.assertTrue(is_prime(17   )   )
         self.assertTrue(is_prime(19   )   )
         self.assertTrue(is_prime(23   )   )
         self.assertTrue(is_prime(29   )   )






      def 			__UpperCamelCase				(	self   )				->Optional[Any]:



         '''simple docstring'''
         with self.assertRaises(lowerCamelCase   ):
            is_prime(-19   )
         self.assertFalse(
             is_prime(0   )			, 'Zero doesn\'t have any positive factors, primes must have exactly two.'			, )
         self.assertFalse(
             is_prime(1   )			, 'One only has 1 positive factor, primes must have exactly two.'			, )
         self.assertFalse(is_prime(2 * 2   )   )
         self.assertFalse(is_prime(2 * 3   )   )
         self.assertFalse(is_prime(3 * 3   )   )
         self.assertFalse(is_prime(3 * 5   )   )
         self.assertFalse(is_prime(3 * 5 * 7   )   )


if __name__ == "__main__":
 unittest.main()







'''simple docstring'''



import os
import re
from shutil import copyfile
from typing import List, Optional, Tuple

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


__UpperCamelCase   :							int         =   logging.get_logger(__name__)

__UpperCamelCase   :							List[str]         =   {
    """vocab_file""": """vocab.txt""",
    """merges_file""": """bpe.codes""",
}

__UpperCamelCase   :							List[str]         =   {
    """vocab_file""": {
        """vinai/phobert-base""": """https://huggingface.co/vinai/phobert-base/resolve/main/vocab.txt""",
        """vinai/phobert-large""": """https://huggingface.co/vinai/phobert-large/resolve/main/vocab.txt""",
    },
    """merges_file""": {
        """vinai/phobert-base""": """https://huggingface.co/vinai/phobert-base/resolve/main/bpe.codes""",
        """vinai/phobert-large""": """https://huggingface.co/vinai/phobert-large/resolve/main/bpe.codes""",
    },
}

__UpperCamelCase   :							Union[str, Any]         =   {
    """vinai/phobert-base""": 256,
    """vinai/phobert-large""": 256,
}



def  __UpperCAmelCase						(					SCREAMING_SNAKE_CASE__:						int     )	->      List[str]:

   """simple docstring"""
   __a					  =     set()
   __a					  =     word[0]
   for char in word[1:]:
      pairs.add((prev_char, char)     )
      __a					  =     char

   __a					  =     set(SCREAMING_SNAKE_CASE__     )
   return pairs



class 				__SCREAMING_SNAKE_CASE							(				_lowerCAmelCase  ):
      __a	       =VOCAB_FILES_NAMES
      __a	       =PRETRAINED_VOCAB_FILES_MAP
      __a	       =PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES




      def __init__(	self			, lowerCamelCase			, lowerCamelCase			, lowerCamelCase="<s>"			, lowerCamelCase="</s>"			, lowerCamelCase="</s>"			, lowerCamelCase="<s>"			, lowerCamelCase="<unk>"			, lowerCamelCase="<pad>"			, lowerCamelCase="<mask>"			, **lowerCamelCase			, )				->str:



         '''simple docstring'''
         super().__init__(
             bos_token=lowerCamelCase			, eos_token=lowerCamelCase			, unk_token=lowerCamelCase			, sep_token=lowerCamelCase			, cls_token=lowerCamelCase			, pad_token=lowerCamelCase			, mask_token=lowerCamelCase			, **lowerCamelCase			, )

         __a					  =     vocab_file
         __a					  =     merges_file

         __a					  =     {}
         __a					  =     0
         __a					  =     1
         __a					  =     2
         __a					  =     3

         self.add_from_file(lowerCamelCase   )

         __a					  =     {v: k for k, v in self.encoder.items()}

         with open(lowerCamelCase			, encoding='utf-8'   ) as merges_handle:
            __a					  =     merges_handle.read().split('\n'   )[:-1]
         __a					  =     [tuple(merge.split()[:-1]   ) for merge in merges]
         __a					  =     dict(zip(lowerCamelCase			, range(len(lowerCamelCase   )   )   )   )
         __a					  =     {}




      def 			__UpperCamelCase				(	self			, lowerCamelCase			, lowerCamelCase = None   )				->List[int]:



         '''simple docstring'''

         if token_ids_a is None:
            return [self.cls_token_id] + token_ids_a + [self.sep_token_id]
         __a					  =     [self.cls_token_id]
         __a					  =     [self.sep_token_id]
         return cls + token_ids_a + sep + sep + token_ids_a + sep




      def 			__UpperCamelCase				(	self			, lowerCamelCase			, lowerCamelCase = None			, lowerCamelCase = False   )				->List[int]:



         '''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   )

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




      def 			__UpperCamelCase				(	self			, lowerCamelCase			, lowerCamelCase = None   )				->List[int]:



         '''simple docstring'''
         __a					  =     [self.sep_token_id]
         __a					  =     [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]




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



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




      def 			__UpperCamelCase				(	self   )				->Optional[int]:



         '''simple docstring'''
         return dict(self.encoder			, **self.added_tokens_encoder   )




      def 			__UpperCamelCase				(	self			, lowerCamelCase   )				->List[str]:



         '''simple docstring'''
         if token in self.cache:
            return self.cache[token]
         __a					  =     tuple(lowerCamelCase   )
         __a					  =     tuple(list(word[:-1]   ) + [word[-1] + '</w>']   )
         __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[:-4]
         __a					  =     word
         return word




      def 			__UpperCamelCase				(	self			, lowerCamelCase   )				->Dict:



         '''simple docstring'''
         __a					  =     []

         __a					  =     re.findall(r'\S+\n?'			, lowerCamelCase   )

         for token in words:
            split_tokens.extend(list(self.bpe(lowerCamelCase   ).split(' '   )   )   )
         return split_tokens




      def 			__UpperCamelCase				(	self			, lowerCamelCase   )				->List[Any]:



         '''simple docstring'''
         return self.encoder.get(lowerCamelCase			, self.encoder.get(self.unk_token   )   )




      def 			__UpperCamelCase				(	self			, lowerCamelCase   )				->List[Any]:



         '''simple docstring'''
         return self.decoder.get(lowerCamelCase			, self.unk_token   )




      def 			__UpperCamelCase				(	self			, lowerCamelCase   )				->Dict:



         '''simple docstring'''
         __a					  =     ' '.join(lowerCamelCase   ).replace('@@ '			, ''   ).strip()
         return out_string




      def 			__UpperCamelCase				(	self			, lowerCamelCase			, lowerCamelCase = 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']   )

         if os.path.abspath(self.vocab_file   ) != os.path.abspath(lowerCamelCase   ):
            copyfile(self.vocab_file			, lowerCamelCase   )

         if os.path.abspath(self.merges_file   ) != os.path.abspath(lowerCamelCase   ):
            copyfile(self.merges_file			, lowerCamelCase   )

         return out_vocab_file, out_merge_file






      def 			__UpperCamelCase				(	self			, lowerCamelCase   )				->str:



         '''simple docstring'''
         if isinstance(lowerCamelCase			, lowerCamelCase   ):
            try:
               with open(lowerCamelCase			, 'r'			, encoding='utf-8'   ) as fd:
                  self.add_from_file(lowerCamelCase   )
            except FileNotFoundError as fnfe:
               raise fnfe
            except UnicodeError:
               raise Exception(F"""Incorrect encoding detected in {f}, please rebuild the dataset"""   )
            return

         __a					  =     f.readlines()
         for lineTmp in lines:
            __a					  =     lineTmp.strip()
            __a					  =     line.rfind(' '   )
            if idx == -1:
               raise ValueError('Incorrect dictionary format, expected \'<token> <cnt>\''   )
            __a					  =     line[:idx]
            __a					  =     len(self.encoder   )