



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


_snake_case	=						logging.get_logger(__name__)

_snake_case	=						{
    "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 						UpperCAmelCase_					( __lowerCAmelCase):
       lowerCamelCase__         =	'decision_transformer'
       lowerCamelCase__         =	['past_key_values']
       lowerCamelCase__         =	{
           'max_position_embeddings': 'n_positions',
           'num_attention_heads': 'n_head',
           'num_hidden_layers': 'n_layer',
       }







       def __init__(     self,    __a=17,    __a=4,    __a=128,    __a=4096,    __a=True,    __a=1,    __a=1024,    __a=3,    __a=1,    __a=None,    __a="relu",    __a=0.1,    __a=0.1,    __a=0.1,    __a=1E-5,    __a=0.02,    __a=True,    __a=True,    __a=5_0256,    __a=5_0256,    __a=False,    __a=False,    **__a,    ):

              '''simple docstring'''

              _lowerCAmelCase  :							Union[str, Any]				     =					state_dim
              _lowerCAmelCase  :							str				     =					act_dim
              _lowerCAmelCase  :							Optional[Any]				     =					hidden_size
              _lowerCAmelCase  :							Any				     =					max_ep_len
              _lowerCAmelCase  :							Union[str, Any]				     =					action_tanh
              _lowerCAmelCase  :							Optional[Any]				     =					vocab_size
              _lowerCAmelCase  :							Dict				     =					n_positions
              _lowerCAmelCase  :							Dict				     =					n_layer
              _lowerCAmelCase  :							str				     =					n_head
              _lowerCAmelCase  :							Union[str, Any]				     =					n_inner
              _lowerCAmelCase  :							Any				     =					activation_function
              _lowerCAmelCase  :							int				     =					resid_pdrop
              _lowerCAmelCase  :							int				     =					embd_pdrop
              _lowerCAmelCase  :							Dict				     =					attn_pdrop
              _lowerCAmelCase  :							List[str]				     =					layer_norm_epsilon
              _lowerCAmelCase  :							Dict				     =					initializer_range
              _lowerCAmelCase  :							Tuple				     =					scale_attn_weights
              _lowerCAmelCase  :							str				     =					use_cache
              _lowerCAmelCase  :							str				     =					scale_attn_by_inverse_layer_idx
              _lowerCAmelCase  :							int				     =					reorder_and_upcast_attn

              _lowerCAmelCase  :							Optional[int]				     =					bos_token_id
              _lowerCAmelCase  :							Dict				     =					eos_token_id

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





def        A       (      _lowerCamelCase       ,				_lowerCamelCase						):





							'''simple docstring'''




							return price * (1 + tax_rate)


if __name__ == "__main__":
						print(f'''{price_plus_tax(100, 0.25) = }''')
						print(f'''{price_plus_tax(125.50, 0.05) = }''')
