'''simple docstring'''




import math
import unittest

def _UpperCamelCase   (       SCREAMING_SNAKE_CASE_							):
     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       UpperCAmelCase__  (       unittest.TestCase			):






   """simple docstring"""
   def 		_lowerCamelCase (self				)    ->     int:
        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 		_lowerCamelCase (self				)    ->     Optional[Any]:
        with self.assertRaises(_a				):
             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'''




from ...configuration_utils import PretrainedConfig
from ...utils import logging


_A								=   logging.get_logger(__name__)

_A								=   {
    'uw-madison/mra-base-512-4': 'https://huggingface.co/uw-madison/mra-base-512-4/resolve/main/config.json',
}


class       UpperCAmelCase__  (       _snake_case			):






   """simple docstring"""
   A							:       Any     =    '''mra'''
   def __init__(self       ,				_a=50_265       ,				_a=768       ,				_a=12       ,				_a=12       ,				_a=3_072       ,				_a="gelu"       ,				_a=0.1       ,				_a=0.1       ,				_a=512       ,				_a=1       ,				_a=0.02       ,				_a=1e-5       ,				_a="absolute"       ,				_a=4       ,				_a="full"       ,				_a=0       ,				_a=0       ,				_a=1       ,				_a=0       ,				_a=2       ,				**_a       ,				)    ->     int:
        super().__init__(pad_token_id=_a       ,				bos_token_id=_a       ,				eos_token_id=_a       ,				**_a				)

        lowercase_					:       Union[str, Any]     				=					vocab_size
        lowercase_					:       List[str]     				=					max_position_embeddings
        lowercase_					:       Optional[Any]     				=					hidden_size
        lowercase_					:       Optional[int]     				=					num_hidden_layers
        lowercase_					:       Dict     				=					num_attention_heads
        lowercase_					:       Optional[int]     				=					intermediate_size
        lowercase_					:       Dict     				=					hidden_act
        lowercase_					:       str     				=					hidden_dropout_prob
        lowercase_					:       int     				=					attention_probs_dropout_prob
        lowercase_					:       str     				=					initializer_range
        lowercase_					:       Union[str, Any]     				=					type_vocab_size
        lowercase_					:       Any     				=					layer_norm_eps
        lowercase_					:       Union[str, Any]     				=					position_embedding_type
        lowercase_					:       Any     				=					block_per_row
        lowercase_					:       Optional[int]     				=					approx_mode
        lowercase_					:       int     				=					initial_prior_first_n_blocks
        lowercase_					:       str     				=					initial_prior_diagonal_n_blocks




