
import inspect
import re

from transformers.utils import direct_transformers_import


# All paths are set with the intent you should run this script from the root of the repo with the command
# python utils/check_config_docstrings.py
UpperCAmelCase	      = '''src/transformers'''


# This is to make sure the transformers module imported is the one in the repo.
UpperCAmelCase	      = direct_transformers_import(PATH_TO_TRANSFORMERS)

UpperCAmelCase	      = transformers.models.auto.configuration_auto.CONFIG_MAPPING

# Regex pattern used to find the checkpoint mentioned in the docstring of `config_class`.
# For example, `[bert-base-uncased](https://huggingface.co/bert-base-uncased)`
UpperCAmelCase	      = re.compile(R'''\[(.+?)\]\((https://huggingface\.co/.+?)\)''')


UpperCAmelCase	      = {
    '''DecisionTransformerConfig''',
    '''EncoderDecoderConfig''',
    '''MusicgenConfig''',
    '''RagConfig''',
    '''SpeechEncoderDecoderConfig''',
    '''TimmBackboneConfig''',
    '''VisionEncoderDecoderConfig''',
    '''VisionTextDualEncoderConfig''',
    '''LlamaConfig''',
}






def 							UpperCAmelCase_		(							__SCREAMING_SNAKE_CASE							):
	lowercase       =    None

	# source code of `config_class`
	lowercase       =    inspect.getsource(_snake_case							)
	lowercase       =    _re_checkpoint.findall(_snake_case							)

	# Each `checkpoint` is a tuple of a checkpoint name and a checkpoint link.
	# For example, `('bert-base-uncased', 'https://huggingface.co/bert-base-uncased')`
	for ckpt_name, ckpt_link in checkpoints:
		# allow the link to end with `/`
		if ckpt_link.endswith('/'							):
			lowercase       =    ckpt_link[:-1]

		# verify the checkpoint name corresponds to the checkpoint link
		lowercase       =    F'''https://huggingface.co/{ckpt_name}'''
		if ckpt_link == ckpt_link_from_name:
			lowercase       =    ckpt_name
			break

	return checkpoint






def 							UpperCAmelCase_		(							):
	lowercase       =    []

	for config_class in list(CONFIG_MAPPING.values()							):
		# Skip deprecated models
		if "models.deprecated" in config_class.__module__:
			continue
		lowercase       =    get_checkpoint_from_config_class(_snake_case							)

		lowercase       =    config_class.__name__
		if checkpoint is None and name not in CONFIG_CLASSES_TO_IGNORE_FOR_DOCSTRING_CHECKPOINT_CHECK:
			configs_without_checkpoint.append(_snake_case							)

	if len(_snake_case							) > 0:
		lowercase       =    '''\n'''.join(sorted(_snake_case							)							)
		raise ValueError(F'''The following configurations don\'t contain any valid checkpoint:\n{message}'''							)


if __name__ == "__main__":
	check_config_docstrings_have_checkpoints()





"""simple docstring"""





import numpy as np


def    _snake_case  (  _snake_case :	np.array	):
      return 1 / (1 + np.exp(-vector	))


if __name__ == "__main__":
 import doctest

 doctest.testmod()


