



import json
from typing import List, Optional, Tuple

from tokenizers import normalizers

from ...tokenization_utils_fast import PreTrainedTokenizerFast
from .tokenization_electra import ElectraTokenizer


UpperCAmelCase__       =	{"vocab_file": "vocab.txt", "tokenizer_file": "tokenizer.json"}

UpperCAmelCase__       =	{
    "vocab_file": {
        "google/electra-small-generator": (
            "https://huggingface.co/google/electra-small-generator/resolve/main/vocab.txt"
        ),
        "google/electra-base-generator": "https://huggingface.co/google/electra-base-generator/resolve/main/vocab.txt",
        "google/electra-large-generator": (
            "https://huggingface.co/google/electra-large-generator/resolve/main/vocab.txt"
        ),
        "google/electra-small-discriminator": (
            "https://huggingface.co/google/electra-small-discriminator/resolve/main/vocab.txt"
        ),
        "google/electra-base-discriminator": (
            "https://huggingface.co/google/electra-base-discriminator/resolve/main/vocab.txt"
        ),
        "google/electra-large-discriminator": (
            "https://huggingface.co/google/electra-large-discriminator/resolve/main/vocab.txt"
        ),
    },
    "tokenizer_file": {
        "google/electra-small-generator": (
            "https://huggingface.co/google/electra-small-generator/resolve/main/tokenizer.json"
        ),
        "google/electra-base-generator": (
            "https://huggingface.co/google/electra-base-generator/resolve/main/tokenizer.json"
        ),
        "google/electra-large-generator": (
            "https://huggingface.co/google/electra-large-generator/resolve/main/tokenizer.json"
        ),
        "google/electra-small-discriminator": (
            "https://huggingface.co/google/electra-small-discriminator/resolve/main/tokenizer.json"
        ),
        "google/electra-base-discriminator": (
            "https://huggingface.co/google/electra-base-discriminator/resolve/main/tokenizer.json"
        ),
        "google/electra-large-discriminator": (
            "https://huggingface.co/google/electra-large-discriminator/resolve/main/tokenizer.json"
        ),
    },
}

UpperCAmelCase__       =	{
    "google/electra-small-generator": 512,
    "google/electra-base-generator": 512,
    "google/electra-large-generator": 512,
    "google/electra-small-discriminator": 512,
    "google/electra-base-discriminator": 512,
    "google/electra-large-discriminator": 512,
}

UpperCAmelCase__       =	{
    "google/electra-small-generator": {"do_lower_case": True},
    "google/electra-base-generator": {"do_lower_case": True},
    "google/electra-large-generator": {"do_lower_case": True},
    "google/electra-small-discriminator": {"do_lower_case": True},
    "google/electra-base-discriminator": {"do_lower_case": True},
    "google/electra-large-discriminator": {"do_lower_case": True},
}







class       __lowerCAmelCase   (      A						):
			UpperCamelCase        =					VOCAB_FILES_NAMES
			UpperCamelCase        =					PRETRAINED_VOCAB_FILES_MAP
			UpperCamelCase        =					PRETRAINED_INIT_CONFIGURATION
			UpperCamelCase        =					PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
			UpperCamelCase        =					ElectraTokenizer



			def __init__(						self       :  Tuple			,		A       :  Dict=None			,		A       :  str=None			,		A       :  Dict=True			,		A       :  Any="[UNK]"			,		A       :  Optional[Any]="[SEP]"			,		A       :  Tuple="[PAD]"			,		A       :  Dict="[CLS]"			,		A       :  List[Any]="[MASK]"			,		A       :  Tuple=True			,		A       :  int=None			,		**A       :  Tuple			,		)       ->   Any:



						"""simple docstring"""






						super().__init__(
						    A			,		tokenizer_file=A			,		do_lower_case=A			,		unk_token=A			,		sep_token=A			,		pad_token=A			,		cls_token=A			,		mask_token=A			,		tokenize_chinese_chars=A			,		strip_accents=A			,		**A			,		)

						_UpperCAmelCase			=       json.loads(self.backend_tokenizer.normalizer.__getstate__())
						if (
						    normalizer_state.get('lowercase'			,		A) != do_lower_case
						    or normalizer_state.get('strip_accents'			,		A) != strip_accents
						    or normalizer_state.get('handle_chinese_chars'			,		A) != tokenize_chinese_chars
						):
									_UpperCAmelCase			=       getattr(A			,		normalizer_state.pop('type'))
									_UpperCAmelCase			=       do_lower_case
									_UpperCAmelCase			=       strip_accents
									_UpperCAmelCase			=       tokenize_chinese_chars
									_UpperCAmelCase			=       normalizer_class(**A)

						_UpperCAmelCase			=       do_lower_case



			def    _lowerCamelCase	(						self       :  Dict			,		A       :  str			,		A       :  Union[str, Any]=None)       ->   str:



						"""simple docstring"""






						_UpperCAmelCase			=       [self.cls_token_id] + token_ids_a + [self.sep_token_id]

						if token_ids_a:
									output += token_ids_a + [self.sep_token_id]

						return output



			def    _lowerCamelCase	(						self       :  str			,		A       :  List[int]			,		A       :  Optional[List[int]] = None)       ->   List[int]:



						"""simple docstring"""






						_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) * [0] + len(token_ids_a + sep) * [1]



			def    _lowerCamelCase	(						self       :  List[Any]			,		A       :  str			,		A       :  Optional[str] = None)       ->   Tuple[str]:



						"""simple docstring"""






						_UpperCAmelCase			=       self._tokenizer.model.save(A			,		name=A)
						return tuple(A)





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


UpperCAmelCase__       =	logging.get_logger(__name__)

UpperCAmelCase__       =	{
    "edbeeching/decision-transformer-gym-hopper-medium": (
        "https://huggingface.co/edbeeching/decision-transformer-gym-hopper-medium/resolve/main/config.json"
    ),
    # See all DecisionTransformer models at https://huggingface.co/models?filter=decision_transformer
}







class       __lowerCAmelCase   (      A						):
			UpperCamelCase        =					'''decision_transformer'''
			UpperCamelCase        =					['''past_key_values''']
			UpperCamelCase        =					{
			    '''max_position_embeddings''': '''n_positions''',
			    '''num_attention_heads''': '''n_head''',
			    '''num_hidden_layers''': '''n_layer''',
			}



			def __init__(						self       :  Any			,		A       :  Optional[int]=17			,		A       :  List[str]=4			,		A       :  int=1_28			,		A       :  Union[str, Any]=40_96			,		A       :  Any=True			,		A       :  Any=1			,		A       :  List[Any]=10_24			,		A       :  List[Any]=3			,		A       :  Tuple=1			,		A       :  Any=None			,		A       :  Optional[int]="relu"			,		A       :  Union[str, Any]=0.1			,		A       :  Optional[int]=0.1			,		A       :  Optional[int]=0.1			,		A       :  Optional[int]=1E-5			,		A       :  List[Any]=0.0_2			,		A       :  Tuple=True			,		A       :  Union[str, Any]=True			,		A       :  str=5_02_56			,		A       :  Union[str, Any]=5_02_56			,		A       :  List[Any]=False			,		A       :  Optional[int]=False			,		**A       :  int			,		)       ->   Union[str, Any]:



						"""simple docstring"""






						_UpperCAmelCase			=       state_dim
						_UpperCAmelCase			=       act_dim
						_UpperCAmelCase			=       hidden_size
						_UpperCAmelCase			=       max_ep_len
						_UpperCAmelCase			=       action_tanh
						_UpperCAmelCase			=       vocab_size
						_UpperCAmelCase			=       n_positions
						_UpperCAmelCase			=       n_layer
						_UpperCAmelCase			=       n_head
						_UpperCAmelCase			=       n_inner
						_UpperCAmelCase			=       activation_function
						_UpperCAmelCase			=       resid_pdrop
						_UpperCAmelCase			=       embd_pdrop
						_UpperCAmelCase			=       attn_pdrop
						_UpperCAmelCase			=       layer_norm_epsilon
						_UpperCAmelCase			=       initializer_range
						_UpperCAmelCase			=       scale_attn_weights
						_UpperCAmelCase			=       use_cache
						_UpperCAmelCase			=       scale_attn_by_inverse_layer_idx
						_UpperCAmelCase			=       reorder_and_upcast_attn

						_UpperCAmelCase			=       bos_token_id
						_UpperCAmelCase			=       eos_token_id

						super().__init__(bos_token_id=A			,		eos_token_id=A			,		**A)
