
from dataclasses import asdict, dataclass
from typing import Optional

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


lowercase__		:str   =    logging.get_logger(__name__)

# TODO Update this
lowercase__		:List[str]   =    {
    "facebook/esm-1b": "https://huggingface.co/facebook/esm-1b/resolve/main/config.json",
    # See all ESM models at https://huggingface.co/models?filter=esm
}







class 				lowercase		(					SCREAMING_SNAKE_CASE__		):
							lowercase_       :       Union[str, Any]       					='''esm'''




							def __init__(				self   ,A__=None   ,A__=None   ,A__=None   ,A__=7_6_8   ,A__=1_2   ,A__=1_2   ,A__=3_0_7_2   ,A__=0.1   ,A__=0.1   ,A__=1_0_2_6   ,A__=0.02   ,A__=1E-12   ,A__="absolute"   ,A__=True   ,A__=None   ,A__=False   ,A__=False   ,A__=None   ,A__=None   ,**A__   ,):
												super().__init__(pad_token_id=A__   ,mask_token_id=A__   ,**A__)

												lowercase							=						vocab_size
												lowercase							=						hidden_size
												lowercase							=						num_hidden_layers
												lowercase							=						num_attention_heads
												lowercase							=						intermediate_size
												lowercase							=						hidden_dropout_prob
												lowercase							=						attention_probs_dropout_prob
												lowercase							=						max_position_embeddings
												lowercase							=						initializer_range
												lowercase							=						layer_norm_eps
												lowercase							=						position_embedding_type
												lowercase							=						use_cache
												lowercase							=						emb_layer_norm_before
												lowercase							=						token_dropout
												lowercase							=						is_folding_model
												if is_folding_model:
																	if esmfold_config is None:
																						logger.info('''No esmfold_config supplied for folding model, using default values.''')
																						lowercase							=						EsmFoldConfig()
																	elif isinstance(A__   ,A__):
																						lowercase							=						EsmFoldConfig(**A__)
																	lowercase							=						esmfold_config
																	if vocab_list is None:
																						logger.warning('''No vocab_list supplied for folding model, assuming the ESM-2 vocabulary!''')
																						lowercase							=						get_default_vocab_list()
																	else:
																						lowercase							=						vocab_list
												else:
																	lowercase							=						None
																	lowercase							=						None
												if self.esmfold_config is not None and getattr(self.esmfold_config   ,'''use_esm_attn_map'''   ,A__):
																	raise ValueError('''The HuggingFace port of ESMFold does not support use_esm_attn_map at this time!''')





							def  A__      (				self):
												lowercase							=						super().to_dict()
												if isinstance(self.esmfold_config   ,A__):
																	lowercase							=						self.esmfold_config.to_dict()
												return output







@dataclass
class 				lowercase		:
							lowercase_       :       str       					=None
							lowercase_       :       bool       					=True
							lowercase_       :       bool       					=False
							lowercase_       :       bool       					=False
							lowercase_       :       bool       					=False
							lowercase_       :       float       					=0

							lowercase_       :       bool       					=True
							lowercase_       :       bool       					=False

							lowercase_       :       int       					=128
							lowercase_       :       "TrunkConfig"       					=None




							def  A__      (				self):
												if self.trunk is None:
																	lowercase							=						TrunkConfig()
												elif isinstance(self.trunk   ,A__):
																	lowercase							=						TrunkConfig(**self.trunk)





							def  A__      (				self):
												lowercase							=						asdict(self)
												lowercase							=						self.trunk.to_dict()
												return output







@dataclass
class 				lowercase		:
							lowercase_       :       int       					=48
							lowercase_       :       int       					=1024
							lowercase_       :       int       					=128
							lowercase_       :       int       					=32
							lowercase_       :       int       					=32
							lowercase_       :       int       					=32
							lowercase_       :       float       					=0
							lowercase_       :       float       					=0
							lowercase_       :       bool       					=False
							lowercase_       :       int       					=4
							lowercase_       :       Optional[int]       					=128
							lowercase_       :       "StructureModuleConfig"       					=None




							def  A__      (				self):
												if self.structure_module is None:
																	lowercase							=						StructureModuleConfig()
												elif isinstance(self.structure_module   ,A__):
																	lowercase							=						StructureModuleConfig(**self.structure_module)

												if self.max_recycles <= 0:
																	raise ValueError(f'`max_recycles` should be positive, got {self.max_recycles}.')
												if self.sequence_state_dim % self.sequence_state_dim != 0:
																	raise ValueError(
																	    '''`sequence_state_dim` should be a round multiple of `sequence_state_dim`, got'''
																	    f' {self.sequence_state_dim} and {self.sequence_state_dim}.')
												if self.pairwise_state_dim % self.pairwise_state_dim != 0:
																	raise ValueError(
																	    '''`pairwise_state_dim` should be a round multiple of `pairwise_state_dim`, got'''
																	    f' {self.pairwise_state_dim} and {self.pairwise_state_dim}.')

												lowercase							=						self.sequence_state_dim // self.sequence_head_width
												lowercase							=						self.pairwise_state_dim // self.pairwise_head_width

												if self.sequence_state_dim != sequence_num_heads * self.sequence_head_width:
																	raise ValueError(
																	    '''`sequence_state_dim` should be equal to `sequence_num_heads * sequence_head_width, got'''
																	    f' {self.sequence_state_dim} != {sequence_num_heads} * {self.sequence_head_width}.')
												if self.pairwise_state_dim != pairwise_num_heads * self.pairwise_head_width:
																	raise ValueError(
																	    '''`pairwise_state_dim` should be equal to `pairwise_num_heads * pairwise_head_width, got'''
																	    f' {self.pairwise_state_dim} != {pairwise_num_heads} * {self.pairwise_head_width}.')
												if self.pairwise_state_dim % 2 != 0:
																	raise ValueError(f'`pairwise_state_dim` should be even, got {self.pairwise_state_dim}.')

												if self.dropout >= 0.4:
																	raise ValueError(f'`dropout` should not be greater than 0.4, got {self.dropout}.')





							def  A__      (				self):
												lowercase							=						asdict(self)
												lowercase							=						self.structure_module.to_dict()
												return output







@dataclass
class 				lowercase		:
							lowercase_       :       int       					=384
							lowercase_       :       int       					=128
							lowercase_       :       int       					=16
							lowercase_       :       int       					=128
							lowercase_       :       int       					=12
							lowercase_       :       int       					=4
							lowercase_       :       int       					=8
							lowercase_       :       float       					=0.1
							lowercase_       :       int       					=8
							lowercase_       :       int       					=1
							lowercase_       :       int       					=2
							lowercase_       :       int       					=7
							lowercase_       :       int       					=10
							lowercase_       :       float       					=1e-8
							lowercase_       :       float       					=1e5




							def  A__      (				self):
												return asdict(self)



def        UpperCamelCase    (							):
					'''simple docstring'''
					return (
					    "<cls>",
					    "<pad>",
					    "<eos>",
					    "<unk>",
					    "L",
					    "A",
					    "G",
					    "V",
					    "S",
					    "E",
					    "R",
					    "T",
					    "I",
					    "D",
					    "P",
					    "K",
					    "Q",
					    "N",
					    "F",
					    "Y",
					    "M",
					    "H",
					    "W",
					    "C",
					    "X",
					    "B",
					    "U",
					    "Z",
					    "O",
					    ".",
					    "-",
					    "<null_1>",
					    "<mask>",
					)






def        UpperCamelCase    (							lowerCAmelCase__		):
					'''simple docstring'''

					if edge <= 0 or not isinstance(lowerCAmelCase__ ,			lowerCAmelCase__		):
										raise ValueError('''Length must be a positive.'''		)
					return 3 * ((25 + 10 * (5 ** (1 / 2))) ** (1 / 2)) * (edge**2)



def        UpperCamelCase    (							lowerCAmelCase__		):
					'''simple docstring'''

					if edge <= 0 or not isinstance(lowerCAmelCase__ ,			lowerCAmelCase__		):
										raise ValueError('''Length must be a positive.'''		)
					return ((15 + (7 * (5 ** (1 / 2)))) / 4) * (edge**3)


if __name__ == "__main__":
							import doctest

							doctest.testmod()




