






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 		lowercase_					(	lowercase   ):






      '''simple docstring'''




      __snake_case       =  '''decision_transformer'''
      __snake_case       =  ['''past_key_values''']
      __snake_case       =  {
          '''max_position_embeddings''': '''n_positions''',
          '''num_attention_heads''': '''n_head''',
          '''num_hidden_layers''': '''n_layer''',
      }







      def __init__(      self      :						Any ,					__UpperCAmelCase      :						List[Any]=17 ,					__UpperCAmelCase      :						int=4 ,					__UpperCAmelCase      :						Optional[int]=128 ,					__UpperCAmelCase      :						Union[str, Any]=4_096 ,					__UpperCAmelCase      :						Optional[int]=True ,					__UpperCAmelCase      :						Optional[Any]=1 ,					__UpperCAmelCase      :						Optional[Any]=1_024 ,					__UpperCAmelCase      :						Any=3 ,					__UpperCAmelCase      :						Dict=1 ,					__UpperCAmelCase      :						List[str]=None ,					__UpperCAmelCase      :						List[str]="relu" ,					__UpperCAmelCase      :						Tuple=0.1 ,					__UpperCAmelCase      :						Any=0.1 ,					__UpperCAmelCase      :						Dict=0.1 ,					__UpperCAmelCase      :						Tuple=1e-5 ,					__UpperCAmelCase      :						str=0.02 ,					__UpperCAmelCase      :						Any=True ,					__UpperCAmelCase      :						Union[str, Any]=True ,					__UpperCAmelCase      :						Optional[int]=50_256 ,					__UpperCAmelCase      :						int=50_256 ,					__UpperCAmelCase      :						Optional[Any]=False ,					__UpperCAmelCase      :						Tuple=False ,					**__UpperCAmelCase      :						List[str] ,					)    ->str:




       """simple docstring"""


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

       a		  =      bos_token_id
       a		  =      eos_token_id

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










def 						_a    (      a :int = 100 )							->			int:
 a		  =      n * (n + 1) * (2 * n + 1) / 6
 a		  =      (n * (n + 1) / 2) ** 2
 return int(square_of_sum - sum_of_squares )


if __name__ == "__main__":
     print(f"""{solution() = }""")


