






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


lowerCamelCase : int						   =       logging.get_logger(__name__)

lowerCamelCase : Optional[int]						   =       {
    "CarlCochet/trajectory-transformer-halfcheetah-medium-v2": (
        "https://huggingface.co/CarlCochet/trajectory-transformer-halfcheetah-medium-v2/resolve/main/config.json"
    ),
    # See all TrajectoryTransformer models at https://huggingface.co/models?filter=trajectory_transformer
}

class 		lowerCAmelCase     ( UpperCAmelCase_			):

					'''simple docstring'''

					_A						:			Optional[int]									=      'trajectory_transformer'
					_A						:			int									=      ['past_key_values']
					_A						:			Optional[Any]									=      {
					    'hidden_size': 'n_embd',
					    'num_attention_heads': 'n_head',
					    'num_hidden_layers': 'n_layer',
					}



					def __init__(  self				:	Union[str, Any]		,     __a				:	List[Any]=100		,     __a				:	List[str]=5		,     __a				:	str=1		,     __a				:	Optional[Any]=1		,     __a				:	Tuple=249		,     __a				:	List[Any]=6		,     __a				:	int=17		,     __a				:	int=25		,     __a				:	str=4		,     __a				:	Union[str, Any]=4		,     __a				:	Optional[Any]=128		,     __a				:	List[str]=0.1		,     __a				:	Dict=0.1		,     __a				:	str=0.1		,     __a				:	int=0.0006		,     __a				:	Optional[int]=512		,     __a				:	Tuple=0.02		,     __a				:	Tuple=1E-12		,     __a				:	int=1		,     __a				:	Any=True		,     __a				:	Any=1		,     __a				:	str=50256		,     __a				:	Optional[Any]=50256		,     **__a				:	Dict		,     )			->  Tuple:


												"""simple docstring"""

												__lowercase      :							Optional[Any]				=     vocab_size
												__lowercase      :							Union[str, Any]				=     action_weight
												__lowercase      :							Any				=     reward_weight
												__lowercase      :							str				=     value_weight
												__lowercase      :							List[str]				=     max_position_embeddings
												__lowercase      :							Dict				=     block_size
												__lowercase      :							Union[str, Any]				=     action_dim
												__lowercase      :							Tuple				=     observation_dim
												__lowercase      :							Any				=     transition_dim
												__lowercase      :							Optional[int]				=     learning_rate
												__lowercase      :							Optional[int]				=     n_layer
												__lowercase      :							Tuple				=     n_head
												__lowercase      :							int				=     n_embd
												__lowercase      :							List[str]				=     embd_pdrop
												__lowercase      :							Optional[Any]				=     attn_pdrop
												__lowercase      :							Tuple				=     resid_pdrop
												__lowercase      :							List[str]				=     initializer_range
												__lowercase      :							List[str]				=     layer_norm_eps
												__lowercase      :							int				=     kaiming_initializer_range
												__lowercase      :							int				=     use_cache
												super().__init__(pad_token_id=_lowercase		,     bos_token_id=_lowercase		,     eos_token_id=_lowercase		,     **_lowercase							)







def       snake_case_		(     lowerCAmelCase_       :						int       ,       lowerCAmelCase_       :						int      ):
							return int((input_a, input_a).count(0      ) == 0      )
def       snake_case_		(     ):
							assert and_gate(0       ,       0      ) == 0
							assert and_gate(0       ,       1      ) == 0
							assert and_gate(1       ,       0      ) == 0
							assert and_gate(1       ,       1      ) == 1


if __name__ == "__main__":
	test_and_gate()
	print(and_gate(1, 0))
	print(and_gate(0, 0))
	print(and_gate(0, 1))
	print(and_gate(1, 1))