



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
A__	      =       """src/transformers"""


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

A__	      =       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)`
A__	      =       re.compile(R"""\[(.+?)\]\((https://huggingface\.co/.+?)\)""")


A__	      =       {
    """DecisionTransformerConfig""",
    """EncoderDecoderConfig""",
    """MusicgenConfig""",
    """RagConfig""",
    """SpeechEncoderDecoderConfig""",
    """TimmBackboneConfig""",
    """VisionEncoderDecoderConfig""",
    """VisionTextDualEncoderConfig""",
    """LlamaConfig""",
}


def 		_UpperCAmelCase				(      snake_case			):





					"""simple docstring"""

					_lowerCAmelCase						=    None

					# source code of `config_class`
					_lowerCAmelCase						=    inspect.getsource(snake_case			)
					_lowerCAmelCase						=    _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("""/"""			):
															_lowerCAmelCase						=    ckpt_link[:-1]

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

					return checkpoint


def 		_UpperCAmelCase				(      ):





					"""simple docstring"""

					_lowerCAmelCase						=    []

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

										_lowerCAmelCase						=    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:
										_lowerCAmelCase						=    """\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()










import os
import unittest

from huggingface_hub.utils import are_progress_bars_disabled

import transformers.models.bart.tokenization_bart
from transformers import logging
from transformers.testing_utils import CaptureLogger, mockenv, mockenv_context
from transformers.utils.logging import disable_progress_bar, enable_progress_bar

class 		SCREAMING_SNAKE_CASE__				(       unittest.TestCase  ):


      def    a (self       :      Dict      ):





         """simple docstring"""



         __snake_case							       =   logging.get_logger()

         # the current default level is logging.WARNING
         __snake_case							       =   logging.get_verbosity()

         logging.set_verbosity_error()
         self.assertEqual(logger.getEffectiveLevel()				,			logging.get_verbosity()      )

         logging.set_verbosity_warning()
         self.assertEqual(logger.getEffectiveLevel()				,			logging.get_verbosity()      )

         logging.set_verbosity_info()
         self.assertEqual(logger.getEffectiveLevel()				,			logging.get_verbosity()      )

         logging.set_verbosity_debug()
         self.assertEqual(logger.getEffectiveLevel()				,			logging.get_verbosity()      )

         # restore to the original level
         logging.set_verbosity(a__      )


      def    a (self       :      Dict      ):





         """simple docstring"""



         __snake_case							       =   logging.get_verbosity()

         __snake_case							       =   logging.get_logger('''transformers.models.bart.tokenization_bart'''      )
         __snake_case							       =   '''Testing 1, 2, 3'''

         # should be able to log warnings (if default settings weren't overridden by `pytest --log-level-all`)
         if level_origin <= logging.WARNING:
            with CaptureLogger(a__      ) as cl:
               logger.warning(a__      )
            self.assertEqual(cl.out				,			msg + '''\n'''      )

         # this is setting the level for all of `transformers.*` loggers
         logging.set_verbosity_error()

         # should not be able to log warnings
         with CaptureLogger(a__      ) as cl:
            logger.warning(a__      )
         self.assertEqual(cl.out				,			''''''      )

         # should be able to log warnings again
         logging.set_verbosity_warning()
         with CaptureLogger(a__      ) as cl:
            logger.warning(a__      )
         self.assertEqual(cl.out				,			msg + '''\n'''      )

         # restore to the original level
         logging.set_verbosity(a__      )


      @mockenv(TRANSFORMERS_VERBOSITY='''error'''      )
      def    a (self       :      Dict      ):





         """simple docstring"""



         transformers.utils.logging._reset_library_root_logger()
         # this action activates the env var
         __snake_case							       =   logging.get_logger('''transformers.models.bart.tokenization_bart'''      )

         __snake_case							       =   os.getenv('''TRANSFORMERS_VERBOSITY'''				,			a__      )
         __snake_case							       =   logging.log_levels[env_level_str]

         __snake_case							       =   logging.get_verbosity()
         self.assertEqual(
             a__				,			a__				,			f"""TRANSFORMERS_VERBOSITY={env_level_str}/{env_level}, but internal verbosity is {current_level}"""				,			)

         # restore to the original level
         __snake_case							       =   ''''''
         transformers.utils.logging._reset_library_root_logger()


      @mockenv(TRANSFORMERS_VERBOSITY='''super-error'''      )
      def    a (self       :      List[Any]      ):





         """simple docstring"""



         transformers.utils.logging._reset_library_root_logger()
         __snake_case							       =   logging.logging.getLogger()
         with CaptureLogger(a__      ) as cl:
            # this action activates the env var
            logging.get_logger('''transformers.models.bart.tokenization_bart'''      )
         self.assertIn('''Unknown option TRANSFORMERS_VERBOSITY=super-error'''				,			cl.out      )

         # no need to restore as nothing was changed


      def    a (self       :      Any      ):





         """simple docstring"""



         transformers.utils.logging._reset_library_root_logger()

         __snake_case							       =   logging.get_logger('''transformers.models.bart.tokenization_bart'''      )
         __snake_case							       =   '''Testing 1, 2, 3'''

         with mockenv_context(TRANSFORMERS_NO_ADVISORY_WARNINGS='''1'''      ):
            # nothing should be logged as env var disables this method
            with CaptureLogger(a__      ) as cl:
               logger.warning_advice(a__      )
            self.assertEqual(cl.out				,			''''''      )

         with mockenv_context(TRANSFORMERS_NO_ADVISORY_WARNINGS=''''''      ):
            # should log normally as TRANSFORMERS_NO_ADVISORY_WARNINGS is unset
            with CaptureLogger(a__      ) as cl:
               logger.warning_advice(a__      )
            self.assertEqual(cl.out				,			msg + '''\n'''      )






def lowerCamelCase__    (							)      ->				str:
   disable_progress_bar()
   assert are_progress_bars_disabled()

   enable_progress_bar()
   assert not are_progress_bars_disabled()

