"""simple docstring"""





import inspect
import os
import re

from transformers.configuration_utils import PretrainedConfig
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     :		List[Any]											=					"src/transformers"


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

_UpperCamelCase     :		Any											=					transformers.models.auto.configuration_auto.CONFIG_MAPPING

_UpperCamelCase     :		Union[str, Any]											=					{
    # used to compute the property `self.chunk_length`
    "EncodecConfig": ["overlap"],
    # used as `self.bert_model = BertModel(config, ...)`
    "DPRConfig": True,
    # not used in modeling files, but it's an important information
    "FSMTConfig": ["langs"],
    # used internally in the configuration class file
    "GPTNeoConfig": ["attention_types"],
    # used internally in the configuration class file
    "EsmConfig": ["is_folding_model"],
    # used during training (despite we don't have training script for these models yet)
    "Mask2FormerConfig": ["ignore_value"],
    # `ignore_value` used during training (despite we don't have training script for these models yet)
    # `norm` used in conversion script (despite not using in the modeling file)
    "OneFormerConfig": ["ignore_value", "norm"],
    # used during preprocessing and collation, see `collating_graphormer.py`
    "GraphormerConfig": ["spatial_pos_max"],
    # used internally in the configuration class file
    "T5Config": ["feed_forward_proj"],
    # used internally in the configuration class file
    # `tokenizer_class` get default value `T5Tokenizer` intentionally
    "MT5Config": ["feed_forward_proj", "tokenizer_class"],
    "UMT5Config": ["feed_forward_proj", "tokenizer_class"],
    # used internally in the configuration class file
    "LongT5Config": ["feed_forward_proj"],
    # used internally in the configuration class file
    "SwitchTransformersConfig": ["feed_forward_proj"],
    # having default values other than `1e-5` - we can't fix them without breaking
    "BioGptConfig": ["layer_norm_eps"],
    # having default values other than `1e-5` - we can't fix them without breaking
    "GLPNConfig": ["layer_norm_eps"],
    # having default values other than `1e-5` - we can't fix them without breaking
    "SegformerConfig": ["layer_norm_eps"],
    # having default values other than `1e-5` - we can't fix them without breaking
    "CvtConfig": ["layer_norm_eps"],
    # having default values other than `1e-5` - we can't fix them without breaking
    "PerceiverConfig": ["layer_norm_eps"],
    # used internally to calculate the feature size
    "InformerConfig": ["num_static_real_features", "num_time_features"],
    # used internally to calculate the feature size
    "TimeSeriesTransformerConfig": ["num_static_real_features", "num_time_features"],
    # used internally to calculate the feature size
    "AutoformerConfig": ["num_static_real_features", "num_time_features"],
    # used internally to calculate `mlp_dim`
    "SamVisionConfig": ["mlp_ratio"],
    # For (head) training, but so far not implemented
    "ClapAudioConfig": ["num_classes"],
    # Not used, but providing useful information to users
    "SpeechT5HifiGanConfig": ["sampling_rate"],
}


# TODO (ydshieh): Check the failing cases, try to fix them or move some cases to the above block once we are sure
SPECIAL_CASES_TO_ALLOW.update(
    {
        "CLIPSegConfig": True,
        "DeformableDetrConfig": True,
        "DetaConfig": True,
        "DinatConfig": True,
        "DonutSwinConfig": True,
        "EfficientFormerConfig": True,
        "FSMTConfig": True,
        "JukeboxConfig": True,
        "LayoutLMv2Config": True,
        "MaskFormerSwinConfig": True,
        "MT5Config": True,
        "NatConfig": True,
        "OneFormerConfig": True,
        "PerceiverConfig": True,
        "RagConfig": True,
        "SpeechT5Config": True,
        "SwinConfig": True,
        "Swin2SRConfig": True,
        "Swinv2Config": True,
        "SwitchTransformersConfig": True,
        "TableTransformerConfig": True,
        "TapasConfig": True,
        "TransfoXLConfig": True,
        "UniSpeechConfig": True,
        "UniSpeechSatConfig": True,
        "WavLMConfig": True,
        "WhisperConfig": True,
        # TODO: @Arthur (for `alignment_head` and `alignment_layer`)
        "JukeboxPriorConfig": True,
        # TODO: @Younes (for `is_decoder`)
        "Pix2StructTextConfig": True,
    }
)





def        a_  (   _lowerCAmelCase						:				str     ,				_lowerCAmelCase						:				Dict     ,				_lowerCAmelCase						:				List[str]     ,				_lowerCAmelCase						:				List[str]  ):





							'''simple docstring'''

							lowercase__    :	int     					=			False
							for attribute in attributes:
														for modeling_source in source_strings:
																					# check if we can find `config.xxx`, `getattr(config, "xxx", ...)` or `getattr(self.config, "xxx", ...)`
																					if (
																					    f"""config.{attribute}""" in modeling_source
																					    or f"""getattr(config, \"{attribute}\"""" in modeling_source
																					    or f"""getattr(self.config, \"{attribute}\"""" in modeling_source
																					):
																												lowercase__    :	Optional[Any]     					=			True
																					# Deal with multi-line cases
																					elif (
																					    re.search(
																					        Rf"""getattr[ \t\v\n\r\f]*\([ \t\v\n\r\f]*(self\.)?config,[ \t\v\n\r\f]*\"{attribute}\""""     ,				_snake_case     ,				)
																					    is not None
																					):
																												lowercase__    :	Any     					=			True
																					# `SequenceSummary` is called with `SequenceSummary(config)`
																					elif attribute in [
																					    "summary_type",
																					    "summary_use_proj",
																					    "summary_activation",
																					    "summary_last_dropout",
																					    "summary_proj_to_labels",
																					    "summary_first_dropout",
																					]:
																												if "SequenceSummary" in modeling_source:
																																			lowercase__    :	List[str]     					=			True
																					if attribute_used:
																												break
														if attribute_used:
																					break

    # common and important attributes, even if they do not always appear in the modeling files
							lowercase__    :	str     					=			[
							    'bos_index',
							    'eos_index',
							    'pad_index',
							    'unk_index',
							    'mask_index',
							    'image_size',
							    'use_cache',
							    'out_features',
							    'out_indices',
							]
							lowercase__    :	Dict     					=			['encoder_no_repeat_ngram_size']

							# Special cases to be allowed
							lowercase__    :	Any     					=			True
							if not attribute_used:
														lowercase__    :	str     					=			False
														for attribute in attributes:
																					# Allow if the default value in the configuration class is different from the one in `PretrainedConfig`
																					if attribute in ["is_encoder_decoder"] and default_value is True:
																												lowercase__    :	Dict     					=			True
																					elif attribute in ["tie_word_embeddings"] and default_value is False:
																												lowercase__    :	Dict     					=			True

																					# Allow cases without checking the default value in the configuration class
																					elif attribute in attributes_to_allow + attributes_used_in_generation:
																												lowercase__    :	List[str]     					=			True
																					elif attribute.endswith('_token_id'  ):
																												lowercase__    :	int     					=			True

																					# configuration class specific cases
																					if not case_allowed:
																												lowercase__    :	Optional[int]     					=			SPECIAL_CASES_TO_ALLOW.get(config_class.__name__     ,				[]  )
																												lowercase__    :	Tuple     					=			allowed_cases is True or attribute in allowed_cases

							return attribute_used or case_allowed





def        a_  (   _lowerCAmelCase						:				Dict  ):





							'''simple docstring'''

							lowercase__    :	Optional[int]     					=			dict(inspect.signature(config_class.__init__  ).parameters  )
							lowercase__    :	Dict     					=			[x for x in list(signature.keys()  ) if x not in ['self', 'kwargs']]
							lowercase__    :	Any     					=			[signature[param].default for param in parameter_names]

							# If `attribute_map` exists, an attribute can have different names to be used in the modeling files, and as long
							# as one variant is used, the test should pass
							lowercase__    :	Dict     					=			{}
							if len(config_class.attribute_map  ) > 0:
														lowercase__    :	Optional[Any]     					=			{v: k for k, v in config_class.attribute_map.items()}

							# Get the path to modeling source files
							lowercase__    :	str     					=			inspect.getsourcefile(_snake_case  )
							lowercase__    :	Tuple     					=			os.path.dirname(_snake_case  )
							# Let's check against all frameworks: as long as one framework uses an attribute, we are good.
							lowercase__    :	str     					=			[os.path.join(_snake_case     ,				_snake_case  ) for fn in os.listdir(_snake_case  ) if fn.startswith('modeling_'  )]

							# Get the source code strings
							lowercase__    :	int     					=			[]
							for path in modeling_paths:
														if os.path.isfile(_snake_case  ):
																					with open(_snake_case  ) as fp:
																												modeling_sources.append(fp.read()  )

							lowercase__    :	List[Any]     					=			[]
							for config_param, default_value in zip(_snake_case     ,				_snake_case  ):
														# `attributes` here is all the variant names for `config_param`
														lowercase__    :	List[str]     					=			[config_param]
														# some configuration classes have non-empty `attribute_map`, and both names could be used in the
														# corresponding modeling files. As long as one of them appears, it is fine.
														if config_param in reversed_attribute_map:
																					attributes.append(reversed_attribute_map[config_param]  )

														if not check_attribute_being_used(_snake_case     ,				_snake_case     ,				_snake_case     ,				_snake_case  ):
																					unused_attributes.append(attributes[0]  )

							return sorted(_snake_case  )





def        a_  (   ):





							'''simple docstring'''

							lowercase__    :	List[Any]     					=			{}
							for _config_class in list(CONFIG_MAPPING.values()  ):
														# Skip deprecated models
														if "models.deprecated" in _config_class.__module__:
																					continue
														# Some config classes are not in `CONFIG_MAPPING` (e.g. `CLIPVisionConfig`, `Blip2VisionConfig`, etc.)
														lowercase__    :	List[Any]     					=			[
														    cls
														    for name, cls in inspect.getmembers(
														        inspect.getmodule(_config_class  )     ,				lambda _lowerCAmelCase  : inspect.isclass(_snake_case  )
														        and issubclass(_snake_case     ,				_snake_case  )
														        and inspect.getmodule(_snake_case  ) == inspect.getmodule(_config_class  )     ,				)
														]
														for config_class in config_classes_in_module:
																					lowercase__    :	Any     					=			check_config_attributes_being_used(_snake_case  )
																					if len(_snake_case  ) > 0:
																												lowercase__    :	List[Any]     					=			unused_attributes

							if len(_snake_case  ) > 0:
														lowercase__    :	Optional[int]     					=			'The following configuration classes contain unused attributes in the corresponding modeling files:\n'
														for name, attributes in configs_with_unused_attributes.items():
																					error += f"""{name}: {attributes}\n"""

														raise ValueError(_snake_case  )


if __name__ == "__main__":
							check_config_attributes()


"""simple docstring"""





import os
import tempfile
import unittest
from pathlib import Path

from transformers import AutoConfig, is_tf_available
from transformers.testing_utils import require_tf


if is_tf_available():
       import tensorflow as tf

       from transformers import TensorFlowBenchmark, TensorFlowBenchmarkArguments


@require_tf
class   UpperCAmelCase_			(					unittest.TestCase):


      def 	_UpperCAmelCase (     self	,    a							) ->      str:
             for model_result in results.values():
                    for batch_size, sequence_length in zip(model_result['bs']	,    model_result['ss']							):
                           lowercase__    :	str     					=			model_result['result'][batch_size][sequence_length]
                           self.assertIsNotNone(a							)


      def 	_UpperCAmelCase (     self							) ->      int:
             lowercase__    :	Dict     					=			'sshleifer/tiny-gpt2'
             lowercase__    :	int     					=			TensorFlowBenchmarkArguments(
                 models=[MODEL_ID]	,    training=a	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    eager_mode=a	,    multi_process=a	,    )
             lowercase__    :	str     					=			TensorFlowBenchmark(a							)
             lowercase__    :	Optional[int]     					=			benchmark.run()
             self.check_results_dict_not_empty(results.time_inference_result							)
             self.check_results_dict_not_empty(results.memory_inference_result							)


      def 	_UpperCAmelCase (     self							) ->      int:
             lowercase__    :	List[str]     					=			'sgugger/tiny-distilbert-classification'
             lowercase__    :	int     					=			TensorFlowBenchmarkArguments(
                 models=[MODEL_ID]	,    training=a	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    multi_process=a	,    only_pretrain_model=a	,    )
             lowercase__    :	Optional[Any]     					=			TensorFlowBenchmark(a							)
             lowercase__    :	Optional[int]     					=			benchmark.run()
             self.check_results_dict_not_empty(results.time_inference_result							)
             self.check_results_dict_not_empty(results.memory_inference_result							)


      def 	_UpperCAmelCase (     self							) ->      Union[str, Any]:
             lowercase__    :	Optional[int]     					=			'sshleifer/tiny-gpt2'
             lowercase__    :	Union[str, Any]     					=			TensorFlowBenchmarkArguments(
                 models=[MODEL_ID]	,    training=a	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    multi_process=a	,    )
             lowercase__    :	Optional[Any]     					=			TensorFlowBenchmark(a							)
             lowercase__    :	List[str]     					=			benchmark.run()
             self.check_results_dict_not_empty(results.time_inference_result							)
             self.check_results_dict_not_empty(results.memory_inference_result							)


      def 	_UpperCAmelCase (     self							) ->      Optional[Any]:
             lowercase__    :	Any     					=			'sshleifer/tiny-gpt2'
             lowercase__    :	List[Any]     					=			AutoConfig.from_pretrained(a							)
             lowercase__    :	Any     					=			TensorFlowBenchmarkArguments(
                 models=[MODEL_ID]	,    training=a	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    eager_mode=a	,    multi_process=a	,    )
             lowercase__    :	Tuple     					=			TensorFlowBenchmark(a	,    [config]							)
             lowercase__    :	Dict     					=			benchmark.run()
             self.check_results_dict_not_empty(results.time_inference_result							)
             self.check_results_dict_not_empty(results.memory_inference_result							)


      def 	_UpperCAmelCase (     self							) ->      int:
             lowercase__    :	Optional[Any]     					=			'sshleifer/tiny-gpt2'
             lowercase__    :	List[str]     					=			AutoConfig.from_pretrained(a							)
             lowercase__    :	Any     					=			TensorFlowBenchmarkArguments(
                 models=[MODEL_ID]	,    training=a	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    multi_process=a	,    )
             lowercase__    :	List[str]     					=			TensorFlowBenchmark(a	,    [config]							)
             lowercase__    :	Any     					=			benchmark.run()
             self.check_results_dict_not_empty(results.time_inference_result							)
             self.check_results_dict_not_empty(results.memory_inference_result							)


      def 	_UpperCAmelCase (     self							) ->      Tuple:
             lowercase__    :	Optional[Any]     					=			'sshleifer/tiny-gpt2'
             lowercase__    :	Any     					=			TensorFlowBenchmarkArguments(
                 models=[MODEL_ID]	,    training=a	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    multi_process=a	,    )
             lowercase__    :	Optional[Any]     					=			TensorFlowBenchmark(a							)
             lowercase__    :	Tuple     					=			benchmark.run()
             self.check_results_dict_not_empty(results.time_train_result							)
             self.check_results_dict_not_empty(results.memory_train_result							)


      def 	_UpperCAmelCase (     self							) ->      str:
             lowercase__    :	Optional[Any]     					=			'sshleifer/tiny-gpt2'
             lowercase__    :	Optional[int]     					=			AutoConfig.from_pretrained(a							)
             lowercase__    :	str     					=			TensorFlowBenchmarkArguments(
                 models=[MODEL_ID]	,    training=a	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    multi_process=a	,    )
             lowercase__    :	str     					=			TensorFlowBenchmark(a	,    [config]							)
             lowercase__    :	Optional[int]     					=			benchmark.run()
             self.check_results_dict_not_empty(results.time_train_result							)
             self.check_results_dict_not_empty(results.memory_train_result							)


      def 	_UpperCAmelCase (     self							) ->      Optional[Any]:
             lowercase__    :	List[str]     					=			'patrickvonplaten/t5-tiny-random'
             lowercase__    :	Any     					=			AutoConfig.from_pretrained(a							)
             lowercase__    :	List[str]     					=			TensorFlowBenchmarkArguments(
                 models=[MODEL_ID]	,    training=a	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    multi_process=a	,    )
             lowercase__    :	int     					=			TensorFlowBenchmark(a	,    configs=[config]							)
             lowercase__    :	Union[str, Any]     					=			benchmark.run()
             self.check_results_dict_not_empty(results.time_inference_result							)
             self.check_results_dict_not_empty(results.memory_inference_result							)


      @unittest.skipIf(is_tf_available() and len(tf.config.list_physical_devices('GPU'							)							) == 0	,    'Cannot do xla on CPU.'							)
      def 	_UpperCAmelCase (     self							) ->      Any:
             lowercase__    :	Any     					=			'sshleifer/tiny-gpt2'
             lowercase__    :	Optional[Any]     					=			TensorFlowBenchmarkArguments(
                 models=[MODEL_ID]	,    training=a	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    use_xla=a	,    multi_process=a	,    )
             lowercase__    :	Any     					=			TensorFlowBenchmark(a							)
             lowercase__    :	Dict     					=			benchmark.run()
             self.check_results_dict_not_empty(results.time_inference_result							)
             self.check_results_dict_not_empty(results.memory_inference_result							)


      def 	_UpperCAmelCase (     self							) ->      List[str]:
             lowercase__    :	Any     					=			'sshleifer/tiny-gpt2'
             with tempfile.TemporaryDirectory() as tmp_dir:
                    lowercase__    :	List[Any]     					=			TensorFlowBenchmarkArguments(
                        models=[MODEL_ID]	,    inference=a	,    save_to_csv=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    inference_time_csv_file=os.path.join(a	,    'inf_time.csv'							)	,    inference_memory_csv_file=os.path.join(a	,    'inf_mem.csv'							)	,    env_info_csv_file=os.path.join(a	,    'env.csv'							)	,    multi_process=a	,    )
                    lowercase__    :	Union[str, Any]     					=			TensorFlowBenchmark(a							)
                    benchmark.run()
                    self.assertTrue(Path(os.path.join(a	,    'inf_time.csv'							)							).exists()							)
                    self.assertTrue(Path(os.path.join(a	,    'inf_mem.csv'							)							).exists()							)
                    self.assertTrue(Path(os.path.join(a	,    'env.csv'							)							).exists()							)


      def 	_UpperCAmelCase (     self							) ->      Dict:
             lowercase__    :	Tuple     					=			'sshleifer/tiny-gpt2'

             def _check_summary_is_not_empty(a							):
                    self.assertTrue(hasattr(a	,    'sequential'							)							)
                    self.assertTrue(hasattr(a	,    'cumulative'							)							)
                    self.assertTrue(hasattr(a	,    'current'							)							)
                    self.assertTrue(hasattr(a	,    'total'							)							)

             with tempfile.TemporaryDirectory() as tmp_dir:
                    lowercase__    :	Optional[Any]     					=			TensorFlowBenchmarkArguments(
                        models=[MODEL_ID]	,    inference=a	,    sequence_lengths=[8]	,    batch_sizes=[1]	,    log_filename=os.path.join(a	,    'log.txt'							)	,    log_print=a	,    trace_memory_line_by_line=a	,    eager_mode=a	,    multi_process=a	,    )
                    lowercase__    :	Optional[int]     					=			TensorFlowBenchmark(a							)
                    lowercase__    :	Optional[Any]     					=			benchmark.run()
                    _check_summary_is_not_empty(result.inference_summary							)
                    self.assertTrue(Path(os.path.join(a	,    'log.txt'							)							).exists()							)

