"""simple docstring"""







import os
import unittest

from transformers import ElectraTokenizerFast
from transformers.models.electra.tokenization_electra import (
    VOCAB_FILES_NAMES,
    BasicTokenizer,
    ElectraTokenizer,
    WordpieceTokenizer,
    _is_control,
    _is_punctuation,
    _is_whitespace,
)
from transformers.testing_utils import require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin, filter_non_english


@require_tokenizers
class      _lowercase						(				__lowerCamelCase , unittest.TestCase					):
				lowerCamelCase__									=    ElectraTokenizer
				lowerCamelCase__									=    ElectraTokenizerFast
				lowerCamelCase__									=    True
				lowerCamelCase__									=    True
				lowerCamelCase__									=    filter_non_english

				def     A	(       self    :      str       ):


										'''simple docstring'''
										super().setUp()

										UpperCAmelCase__   		=						[
										    """[UNK]""",
										    """[CLS]""",
										    """[SEP]""",
										    """[PAD]""",
										    """[MASK]""",
										    """want""",
										    """##want""",
										    """##ed""",
										    """wa""",
										    """un""",
										    """runn""",
										    """##ing""",
										    """,""",
										    """low""",
										    """lowest""",
										]
										UpperCAmelCase__   		=						os.path.join(self.tmpdirname    ,      VOCAB_FILES_NAMES["""vocab_file"""]       )
										with open(self.vocab_file    ,      """w"""    ,      encoding="""utf-8"""       ) as vocab_writer:
																vocab_writer.write("""""".join([x + """\n""" for x in vocab_tokens]       )       )

				def     A	(       self    :      List[str]    ,      _lowercase    :      str       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						"""UNwant\u00E9d,running"""
										UpperCAmelCase__   		=						"""unwanted, running"""
										return input_text, output_text

				def     A	(       self    :      Tuple       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						self.tokenizer_class(self.vocab_file       )

										UpperCAmelCase__   		=						tokenizer.tokenize("""UNwant\u00E9d,running"""       )
										self.assertListEqual(_lowercase    ,      ["""un""", """##want""", """##ed""", """,""", """runn""", """##ing"""]       )
										self.assertListEqual(tokenizer.convert_tokens_to_ids(_lowercase       )    ,      [9, 6, 7, 1_2, 1_0, 1_1]       )

				def     A	(       self    :      Tuple       ):


										'''simple docstring'''
										if not self.test_rust_tokenizer:
																return

										UpperCAmelCase__   		=						self.get_tokenizer()
										UpperCAmelCase__   		=						self.get_rust_tokenizer()

										UpperCAmelCase__   		=						"""UNwant\u00E9d,running"""

										UpperCAmelCase__   		=						tokenizer.tokenize(_lowercase       )
										UpperCAmelCase__   		=						rust_tokenizer.tokenize(_lowercase       )
										self.assertListEqual(_lowercase    ,      _lowercase       )

										UpperCAmelCase__   		=						tokenizer.encode(_lowercase    ,      add_special_tokens=_lowercase       )
										UpperCAmelCase__   		=						rust_tokenizer.encode(_lowercase    ,      add_special_tokens=_lowercase       )
										self.assertListEqual(_lowercase    ,      _lowercase       )

										UpperCAmelCase__   		=						self.get_rust_tokenizer()
										UpperCAmelCase__   		=						tokenizer.encode(_lowercase       )
										UpperCAmelCase__   		=						rust_tokenizer.encode(_lowercase       )
										self.assertListEqual(_lowercase    ,      _lowercase       )

										# With lower casing
										UpperCAmelCase__   		=						self.get_tokenizer(do_lower_case=_lowercase       )
										UpperCAmelCase__   		=						self.get_rust_tokenizer(do_lower_case=_lowercase       )

										UpperCAmelCase__   		=						"""UNwant\u00E9d,running"""

										UpperCAmelCase__   		=						tokenizer.tokenize(_lowercase       )
										UpperCAmelCase__   		=						rust_tokenizer.tokenize(_lowercase       )
										self.assertListEqual(_lowercase    ,      _lowercase       )

										UpperCAmelCase__   		=						tokenizer.encode(_lowercase    ,      add_special_tokens=_lowercase       )
										UpperCAmelCase__   		=						rust_tokenizer.encode(_lowercase    ,      add_special_tokens=_lowercase       )
										self.assertListEqual(_lowercase    ,      _lowercase       )

										UpperCAmelCase__   		=						self.get_rust_tokenizer()
										UpperCAmelCase__   		=						tokenizer.encode(_lowercase       )
										UpperCAmelCase__   		=						rust_tokenizer.encode(_lowercase       )
										self.assertListEqual(_lowercase    ,      _lowercase       )

				def     A	(       self    :      List[str]       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						BasicTokenizer()

										self.assertListEqual(tokenizer.tokenize("""ah\u535A\u63A8zz"""       )    ,      ["""ah""", """\u535A""", """\u63A8""", """zz"""]       )

				def     A	(       self    :      Tuple       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						BasicTokenizer(do_lower_case=_lowercase       )

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHeLLo!how  \n Are yoU?  """       )    ,      ["""hello""", """!""", """how""", """are""", """you""", """?"""]       )
										self.assertListEqual(tokenizer.tokenize("""H\u00E9llo"""       )    ,      ["""hello"""]       )

				def     A	(       self    :      Any       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						BasicTokenizer(do_lower_case=_lowercase    ,      strip_accents=_lowercase       )

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """       )    ,      ["""hällo""", """!""", """how""", """are""", """you""", """?"""]       )
										self.assertListEqual(tokenizer.tokenize("""H\u00E9llo"""       )    ,      ["""h\u00E9llo"""]       )

				def     A	(       self    :      Optional[Any]       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						BasicTokenizer(do_lower_case=_lowercase    ,      strip_accents=_lowercase       )

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """       )    ,      ["""hallo""", """!""", """how""", """are""", """you""", """?"""]       )
										self.assertListEqual(tokenizer.tokenize("""H\u00E9llo"""       )    ,      ["""hello"""]       )

				def     A	(       self    :      Tuple       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						BasicTokenizer(do_lower_case=_lowercase       )

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """       )    ,      ["""hallo""", """!""", """how""", """are""", """you""", """?"""]       )
										self.assertListEqual(tokenizer.tokenize("""H\u00E9llo"""       )    ,      ["""hello"""]       )

				def     A	(       self    :      Union[str, Any]       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						BasicTokenizer(do_lower_case=_lowercase       )

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHeLLo!how  \n Are yoU?  """       )    ,      ["""HeLLo""", """!""", """how""", """Are""", """yoU""", """?"""]       )

				def     A	(       self    :      Tuple       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						BasicTokenizer(do_lower_case=_lowercase    ,      strip_accents=_lowercase       )

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """       )    ,      ["""HäLLo""", """!""", """how""", """Are""", """yoU""", """?"""]       )

				def     A	(       self    :      Optional[Any]       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						BasicTokenizer(do_lower_case=_lowercase    ,      strip_accents=_lowercase       )

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """       )    ,      ["""HaLLo""", """!""", """how""", """Are""", """yoU""", """?"""]       )

				def     A	(       self    :      Optional[int]       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						BasicTokenizer(do_lower_case=_lowercase    ,      never_split=["""[UNK]"""]       )

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHeLLo!how  \n Are yoU? [UNK]"""       )    ,      ["""HeLLo""", """!""", """how""", """Are""", """yoU""", """?""", """[UNK]"""]       )

				def     A	(       self    :      Union[str, Any]       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						["""[UNK]""", """[CLS]""", """[SEP]""", """want""", """##want""", """##ed""", """wa""", """un""", """runn""", """##ing"""]

										UpperCAmelCase__   		=						{}
										for i, token in enumerate(_lowercase       ):
																UpperCAmelCase__   		=						i
										UpperCAmelCase__   		=						WordpieceTokenizer(vocab=_lowercase    ,      unk_token="""[UNK]"""       )

										self.assertListEqual(tokenizer.tokenize(""""""       )    ,      []       )

										self.assertListEqual(tokenizer.tokenize("""unwanted running"""       )    ,      ["""un""", """##want""", """##ed""", """runn""", """##ing"""]       )

										self.assertListEqual(tokenizer.tokenize("""unwantedX running"""       )    ,      ["""[UNK]""", """runn""", """##ing"""]       )

				def     A	(       self    :      Dict       ):


										'''simple docstring'''
										self.assertTrue(_is_whitespace(""" """       )       )
										self.assertTrue(_is_whitespace("""\t"""       )       )
										self.assertTrue(_is_whitespace("""\r"""       )       )
										self.assertTrue(_is_whitespace("""\n"""       )       )
										self.assertTrue(_is_whitespace("""\u00A0"""       )       )

										self.assertFalse(_is_whitespace("""A"""       )       )
										self.assertFalse(_is_whitespace("""-"""       )       )

				def     A	(       self    :      Union[str, Any]       ):


										'''simple docstring'''
										self.assertTrue(_is_control("""\u0005"""       )       )

										self.assertFalse(_is_control("""A"""       )       )
										self.assertFalse(_is_control(""" """       )       )
										self.assertFalse(_is_control("""\t"""       )       )
										self.assertFalse(_is_control("""\r"""       )       )

				def     A	(       self    :      List[Any]       ):


										'''simple docstring'''
										self.assertTrue(_is_punctuation("""-"""       )       )
										self.assertTrue(_is_punctuation("""$"""       )       )
										self.assertTrue(_is_punctuation("""`"""       )       )
										self.assertTrue(_is_punctuation("""."""       )       )

										self.assertFalse(_is_punctuation("""A"""       )       )
										self.assertFalse(_is_punctuation(""" """       )       )

				def     A	(       self    :      Union[str, Any]       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						self.get_tokenizer()
										UpperCAmelCase__   		=						self.get_rust_tokenizer()

										# Example taken from the issue https://github.com/huggingface/tokenizers/issues/340
										self.assertListEqual([tokenizer.tokenize(_lowercase       ) for t in ["""Test""", """\xad""", """test"""]]    ,      [["""[UNK]"""], [], ["""[UNK]"""]]       )

										self.assertListEqual(
										    [rust_tokenizer.tokenize(_lowercase       ) for t in ["""Test""", """\xad""", """test"""]]    ,      [["""[UNK]"""], [], ["""[UNK]"""]]       )

				@slow
				def     A	(       self    :      int       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						self.tokenizer_class.from_pretrained("""google/electra-base-discriminator"""       )

										UpperCAmelCase__   		=						tokenizer.encode("""sequence builders"""    ,      add_special_tokens=_lowercase       )
										UpperCAmelCase__   		=						tokenizer.encode("""multi-sequence build"""    ,      add_special_tokens=_lowercase       )

										UpperCAmelCase__   		=						tokenizer.build_inputs_with_special_tokens(_lowercase       )
										UpperCAmelCase__   		=						tokenizer.build_inputs_with_special_tokens(_lowercase    ,      _lowercase       )

										assert encoded_sentence == [1_0_1] + text + [1_0_2]
										assert encoded_pair == [1_0_1] + text + [1_0_2] + text_a + [1_0_2]

				def     A	(       self    :      Tuple       ):


										'''simple docstring'''
										for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
																with self.subTest(f'{tokenizer.__class__.__name__} ({pretrained_name})'       ):
																						UpperCAmelCase__   		=						self.rust_tokenizer_class.from_pretrained(_lowercase    ,      **_lowercase       )

																						UpperCAmelCase__   		=						f'A, naïve {tokenizer_r.mask_token} AllenNLP sentence.'
																						UpperCAmelCase__   		=						tokenizer_r.encode_plus(
																						    _lowercase    ,      return_attention_mask=_lowercase    ,      return_token_type_ids=_lowercase    ,      return_offsets_mapping=_lowercase    ,      add_special_tokens=_lowercase    ,      )

																						UpperCAmelCase__   		=						tokenizer_r.do_lower_case if hasattr(_lowercase    ,      """do_lower_case"""       ) else False
																						UpperCAmelCase__   		=						(
																						    [
																						        ((0, 0), tokenizer_r.cls_token),
																						        ((0, 1), """A"""),
																						        ((1, 2), ""","""),
																						        ((3, 5), """na"""),
																						        ((5, 6), """##ï"""),
																						        ((6, 8), """##ve"""),
																						        ((9, 1_5), tokenizer_r.mask_token),
																						        ((1_6, 2_1), """Allen"""),
																						        ((2_1, 2_3), """##NL"""),
																						        ((2_3, 2_4), """##P"""),
																						        ((2_5, 3_3), """sentence"""),
																						        ((3_3, 3_4), """."""),
																						        ((0, 0), tokenizer_r.sep_token),
																						    ]
																						    if not do_lower_case
																						    else [
																						        ((0, 0), tokenizer_r.cls_token),
																						        ((0, 1), """a"""),
																						        ((1, 2), ""","""),
																						        ((3, 8), """naive"""),
																						        ((9, 1_5), tokenizer_r.mask_token),
																						        ((1_6, 2_1), """allen"""),
																						        ((2_1, 2_3), """##nl"""),
																						        ((2_3, 2_4), """##p"""),
																						        ((2_5, 3_3), """sentence"""),
																						        ((3_3, 3_4), """."""),
																						        ((0, 0), tokenizer_r.sep_token),
																						    ]
																						)

																						self.assertEqual(
																						    [e[1] for e in expected_results]    ,      tokenizer_r.convert_ids_to_tokens(tokens["""input_ids"""]       )       )
																						self.assertEqual([e[0] for e in expected_results]    ,      tokens["""offset_mapping"""]       )

				def     A	(       self    :      Optional[Any]       ):


										'''simple docstring'''
										UpperCAmelCase__   		=						["""的""", """人""", """有"""]
										UpperCAmelCase__   		=						"""""".join(_lowercase       )
										for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
																with self.subTest(f'{tokenizer.__class__.__name__} ({pretrained_name})'       ):
																						UpperCAmelCase__   		=						True
																						UpperCAmelCase__   		=						self.tokenizer_class.from_pretrained(_lowercase    ,      **_lowercase       )
																						UpperCAmelCase__   		=						self.rust_tokenizer_class.from_pretrained(_lowercase    ,      **_lowercase       )

																						UpperCAmelCase__   		=						tokenizer_p.encode(_lowercase    ,      add_special_tokens=_lowercase       )
																						UpperCAmelCase__   		=						tokenizer_r.encode(_lowercase    ,      add_special_tokens=_lowercase       )

																						UpperCAmelCase__   		=						tokenizer_r.convert_ids_to_tokens(_lowercase       )
																						UpperCAmelCase__   		=						tokenizer_p.convert_ids_to_tokens(_lowercase       )

																						# it is expected that each Chinese character is not preceded by "##"
																						self.assertListEqual(_lowercase    ,      _lowercase       )
																						self.assertListEqual(_lowercase    ,      _lowercase       )

																						UpperCAmelCase__   		=						False
																						UpperCAmelCase__   		=						self.rust_tokenizer_class.from_pretrained(_lowercase    ,      **_lowercase       )
																						UpperCAmelCase__   		=						self.tokenizer_class.from_pretrained(_lowercase    ,      **_lowercase       )

																						UpperCAmelCase__   		=						tokenizer_r.encode(_lowercase    ,      add_special_tokens=_lowercase       )
																						UpperCAmelCase__   		=						tokenizer_p.encode(_lowercase    ,      add_special_tokens=_lowercase       )

																						UpperCAmelCase__   		=						tokenizer_r.convert_ids_to_tokens(_lowercase       )
																						UpperCAmelCase__   		=						tokenizer_p.convert_ids_to_tokens(_lowercase       )

																						# it is expected that only the first Chinese character is not preceded by "##".
																						UpperCAmelCase__   		=						[
																						    f'##{token}' if idx != 0 else token for idx, token in enumerate(_lowercase       )
																						]
																						self.assertListEqual(_lowercase    ,      _lowercase       )
																						self.assertListEqual(_lowercase    ,      _lowercase       )


"""simple docstring"""







from collections import OrderedDict
from typing import Mapping

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging


UpperCamelCase_    :    Optional[Any]											=						logging.get_logger(__name__)

UpperCamelCase_    :    Optional[int]											=						{
    """google/rembert""": """https://huggingface.co/google/rembert/resolve/main/config.json""",
    # See all RemBERT models at https://huggingface.co/models?filter=rembert
}


class      _lowercase						(				__lowerCamelCase					):
				lowerCamelCase__									=    'rembert'

				def __init__(       self    :      List[Any]    ,      _lowercase    :      Tuple=2_5_0_3_0_0    ,      _lowercase    :      Tuple=1_1_5_2    ,      _lowercase    :      List[Any]=3_2    ,      _lowercase    :      Dict=1_8    ,      _lowercase    :      str=2_5_6    ,      _lowercase    :      Optional[Any]=1_6_6_4    ,      _lowercase    :      Union[str, Any]=4_6_0_8    ,      _lowercase    :      Optional[Any]="gelu"    ,      _lowercase    :      Optional[Any]=0.0    ,      _lowercase    :      Tuple=0.0    ,      _lowercase    :      Dict=0.1    ,      _lowercase    :      str=5_1_2    ,      _lowercase    :      Tuple=2    ,      _lowercase    :      int=0.02    ,      _lowercase    :      Tuple=1e-12    ,      _lowercase    :      Union[str, Any]=True    ,      _lowercase    :      Optional[int]=0    ,      _lowercase    :      Any=3_1_2    ,      _lowercase    :      Optional[Any]=3_1_3    ,      **_lowercase    :      str    ,      ):


										'''simple docstring'''
										super().__init__(pad_token_id=_lowercase    ,      bos_token_id=_lowercase    ,      eos_token_id=_lowercase    ,      **_lowercase       )

										UpperCAmelCase__   		=						vocab_size
										UpperCAmelCase__   		=						input_embedding_size
										UpperCAmelCase__   		=						output_embedding_size
										UpperCAmelCase__   		=						max_position_embeddings
										UpperCAmelCase__   		=						hidden_size
										UpperCAmelCase__   		=						num_hidden_layers
										UpperCAmelCase__   		=						num_attention_heads
										UpperCAmelCase__   		=						intermediate_size
										UpperCAmelCase__   		=						hidden_act
										UpperCAmelCase__   		=						hidden_dropout_prob
										UpperCAmelCase__   		=						attention_probs_dropout_prob
										UpperCAmelCase__   		=						classifier_dropout_prob
										UpperCAmelCase__   		=						initializer_range
										UpperCAmelCase__   		=						type_vocab_size
										UpperCAmelCase__   		=						layer_norm_eps
										UpperCAmelCase__   		=						use_cache
										UpperCAmelCase__   		=						False


class      _lowercase						(				__lowerCamelCase					):

				@property
				def     A	(       self    :      str       ):


										'''simple docstring'''
										if self.task == "multiple-choice":
																UpperCAmelCase__   		=						{0: """batch""", 1: """choice""", 2: """sequence"""}
										else:
																UpperCAmelCase__   		=						{0: """batch""", 1: """sequence"""}
										return OrderedDict(
										    [
										        ("""input_ids""", dynamic_axis),
										        ("""attention_mask""", dynamic_axis),
										        ("""token_type_ids""", dynamic_axis),
										    ]       )

				@property
				def     A	(       self    :      Tuple       ):


										'''simple docstring'''
										return 1e-4

