





'''simple docstring'''

import json
import os
import shutil
import sys
import tempfile
import unittest
import unittest.mock as mock
from pathlib import Path

from huggingface_hub import HfFolder, delete_repo
from requests.exceptions import HTTPError

from transformers import AutoConfig, BertConfig, GPTaConfig
from transformers.configuration_utils import PretrainedConfig
from transformers.testing_utils import TOKEN, USER, is_staging_test


sys.path.append(str(Path(__file__).parent.parent / "utils"))

from test_module.custom_configuration import CustomConfig  # noqa E402


__lowerCamelCase : List[Any]											=			{
    "return_dict": False,
    "output_hidden_states": True,
    "output_attentions": True,
    "torchscript": True,
    "torch_dtype": "float16",
    "use_bfloat16": True,
    "tf_legacy_loss": True,
    "pruned_heads": {"a": 1},
    "tie_word_embeddings": False,
    "is_decoder": True,
    "cross_attention_hidden_size": 128,
    "add_cross_attention": True,
    "tie_encoder_decoder": True,
    "max_length": 50,
    "min_length": 3,
    "do_sample": True,
    "early_stopping": True,
    "num_beams": 3,
    "num_beam_groups": 3,
    "diversity_penalty": 0.5,
    "temperature": 2.0,
    "top_k": 10,
    "top_p": 0.7,
    "typical_p": 0.2,
    "repetition_penalty": 0.8,
    "length_penalty": 0.8,
    "no_repeat_ngram_size": 5,
    "encoder_no_repeat_ngram_size": 5,
    "bad_words_ids": [1, 2, 3],
    "num_return_sequences": 3,
    "chunk_size_feed_forward": 5,
    "output_scores": True,
    "return_dict_in_generate": True,
    "forced_bos_token_id": 2,
    "forced_eos_token_id": 3,
    "remove_invalid_values": True,
    "architectures": ["BertModel"],
    "finetuning_task": "translation",
    "id2label": {0: "label"},
    "label2id": {"label": "0"},
    "tokenizer_class": "BertTokenizerFast",
    "prefix": "prefix",
    "bos_token_id": 6,
    "pad_token_id": 7,
    "eos_token_id": 8,
    "sep_token_id": 9,
    "decoder_start_token_id": 10,
    "exponential_decay_length_penalty": (5, 1.01),
    "suppress_tokens": [0, 1],
    "begin_suppress_tokens": 2,
    "task_specific_params": {"translation": "some_params"},
    "problem_type": "regression",
}

@is_staging_test
class       UpperCAmelCase (						unittest.TestCase  ):




		@classmethod
		def 				UpperCAmelCase__   (cls      :	Tuple				)					-> Dict:
						lowercase         = TOKEN
						HfFolder.save_token(A__				)




		@classmethod
		def 				UpperCAmelCase__   (cls      :	int				)					-> int:
						try:
										delete_repo(token=cls._token							,			repo_id="test-config"				)
						except HTTPError:
										pass

						try:
										delete_repo(token=cls._token							,			repo_id="valid_org/test-config-org"				)
						except HTTPError:
										pass

						try:
										delete_repo(token=cls._token							,			repo_id="test-dynamic-config"				)
						except HTTPError:
										pass




		def 				UpperCAmelCase__   (self      :	int				)					-> str:
						lowercase         = BertConfig(
						    vocab_size=9_9							,			hidden_size=3_2							,			num_hidden_layers=5							,			num_attention_heads=4							,			intermediate_size=3_7				)
						config.push_to_hub("test-config"							,			use_auth_token=self._token				)

						lowercase         = BertConfig.from_pretrained(f'{USER}/test-config'				)
						for k, v in config.to_dict().items():
										if k != "transformers_version":
														self.assertEqual(A__							,			getattr(A__							,			A__				)				)

        # Reset repo
						delete_repo(token=self._token							,			repo_id="test-config"				)

						# Push to hub via save_pretrained
						with tempfile.TemporaryDirectory() as tmp_dir:
										config.save_pretrained(A__							,			repo_id="test-config"							,			push_to_hub=A__							,			use_auth_token=self._token				)

						lowercase         = BertConfig.from_pretrained(f'{USER}/test-config'				)
						for k, v in config.to_dict().items():
										if k != "transformers_version":
														self.assertEqual(A__							,			getattr(A__							,			A__				)				)




		def 				UpperCAmelCase__   (self      :	Union[str, Any]				)					-> List[Any]:
						lowercase         = BertConfig(
						    vocab_size=9_9							,			hidden_size=3_2							,			num_hidden_layers=5							,			num_attention_heads=4							,			intermediate_size=3_7				)
						config.push_to_hub("valid_org/test-config-org"							,			use_auth_token=self._token				)

						lowercase         = BertConfig.from_pretrained("valid_org/test-config-org"				)
						for k, v in config.to_dict().items():
										if k != "transformers_version":
														self.assertEqual(A__							,			getattr(A__							,			A__				)				)

        # Reset repo
						delete_repo(token=self._token							,			repo_id="valid_org/test-config-org"				)

						# Push to hub via save_pretrained
						with tempfile.TemporaryDirectory() as tmp_dir:
										config.save_pretrained(
										    A__							,			repo_id="valid_org/test-config-org"							,			push_to_hub=A__							,			use_auth_token=self._token				)

						lowercase         = BertConfig.from_pretrained("valid_org/test-config-org"				)
						for k, v in config.to_dict().items():
										if k != "transformers_version":
														self.assertEqual(A__							,			getattr(A__							,			A__				)				)





		def 				UpperCAmelCase__   (self      :	Union[str, Any]				)					-> List[str]:
						CustomConfig.register_for_auto_class()
						lowercase         = CustomConfig(attribute=4_2				)

						config.push_to_hub("test-dynamic-config"							,			use_auth_token=self._token				)

						# This has added the proper auto_map field to the config
						self.assertDictEqual(config.auto_map							,			{"AutoConfig": "custom_configuration.CustomConfig"}				)

						lowercase         = AutoConfig.from_pretrained(f'{USER}/test-dynamic-config'							,			trust_remote_code=A__				)
						# Can't make an isinstance check because the new_config is from the FakeConfig class of a dynamic module
						self.assertEqual(new_config.__class__.__name__							,			"CustomConfig"				)
						self.assertEqual(new_config.attribute							,			4_2				)




class       UpperCAmelCase (						unittest.TestCase  ):




		def 				UpperCAmelCase__   (self      :	Union[str, Any]				)					-> Optional[int]:
						lowercase         = GPTaConfig()

						# attempt to modify each of int/float/bool/str config records and verify they were updated
						lowercase         = c.n_embd + 1  # int
						lowercase         = c.resid_pdrop + 1.0  # float
						lowercase         = not c.scale_attn_weights  # bool
						lowercase         = c.summary_type + "foo"  # str
						c.update_from_string(
						    f'n_embd={n_embd},resid_pdrop={resid_pdrop},scale_attn_weights={scale_attn_weights},summary_type={summary_type}'				)
						self.assertEqual(A__							,			c.n_embd							,			"mismatch for key: n_embd"				)
						self.assertEqual(A__							,			c.resid_pdrop							,			"mismatch for key: resid_pdrop"				)
						self.assertEqual(A__							,			c.scale_attn_weights							,			"mismatch for key: scale_attn_weights"				)
						self.assertEqual(A__							,			c.summary_type							,			"mismatch for key: summary_type"				)




		def 				UpperCAmelCase__   (self      :	str				)					-> Union[str, Any]:
						lowercase         = PretrainedConfig()
						lowercase         = [key for key in base_config.__dict__ if key not in config_common_kwargs]
						# If this part of the test fails, you have arguments to addin config_common_kwargs above.
						self.assertListEqual(
						    A__							,			["is_encoder_decoder", "_name_or_path", "_commit_hash", "transformers_version"]				)
						lowercase         = [key for key, value in config_common_kwargs.items() if value == getattr(A__							,			A__				)]
						if len(A__				) > 0:
										raise ValueError(
										    "The following keys are set with the default values in"
										    " `test_configuration_common.config_common_kwargs` pick another value for them:"
										    f' {", ".join(A__				)}.'				)




		def 				UpperCAmelCase__   (self      :	Tuple				)					-> Optional[Any]:
						with self.assertRaises(A__				):
										# config is in subfolder, the following should not work without specifying the subfolder
										lowercase         = BertConfig.from_pretrained("hf-internal-testing/tiny-random-bert-subfolder"				)

						lowercase         = BertConfig.from_pretrained("hf-internal-testing/tiny-random-bert-subfolder"							,			subfolder="bert"				)

						self.assertIsNotNone(A__				)




		def 				UpperCAmelCase__   (self      :	int				)					-> List[Any]:
						# A mock response for an HTTP head request to emulate server down
						lowercase         = mock.Mock()
						lowercase         = 5_0_0
						lowercase         = {}
						lowercase         = HTTPError
						lowercase         = {}

						# Download this model to make sure it's in the cache.
						lowercase         = BertConfig.from_pretrained("hf-internal-testing/tiny-random-bert"				)

						# Under the mock environment we get a 500 error when trying to reach the model.
						with mock.patch("requests.Session.request"							,			return_value=A__				) as mock_head:
										lowercase         = BertConfig.from_pretrained("hf-internal-testing/tiny-random-bert"				)
										# This check we did call the fake head request
										mock_head.assert_called()




		def 				UpperCAmelCase__   (self      :	str				)					-> str:
						# This test is for deprecated behavior and can be removed in v5
						lowercase         = BertConfig.from_pretrained(
						    "https://huggingface.co/hf-internal-testing/tiny-random-bert/resolve/main/config.json"				)




		def 				UpperCAmelCase__   (self      :	str				)					-> int:
						lowercase         = AutoConfig.from_pretrained("bert-base-cased"				)
						lowercase         = ["config.4.0.0.json"]

						with tempfile.TemporaryDirectory() as tmp_dir:
										configuration.save_pretrained(A__				)
										lowercase         = 2
										json.dump(configuration.to_dict()							,			open(os.path.join(A__							,			"config.4.0.0.json"				)							,			"w"				)				)

										# This should pick the new configuration file as the version of Transformers is > 4.0.0
										lowercase         = AutoConfig.from_pretrained(A__				)
										self.assertEqual(new_configuration.hidden_size							,			2				)

										# Will need to be adjusted if we reach v42 and this test is still here.
										# Should pick the old configuration file as the version of Transformers is < 4.42.0
										lowercase         = ["config.42.0.0.json"]
										lowercase         = 7_6_8
										configuration.save_pretrained(A__				)
										shutil.move(os.path.join(A__							,			"config.4.0.0.json"				)							,			os.path.join(A__							,			"config.42.0.0.json"				)				)
										lowercase         = AutoConfig.from_pretrained(A__				)
										self.assertEqual(new_configuration.hidden_size							,			7_6_8				)





		def 				UpperCAmelCase__   (self      :	Optional[Any]				)					-> Dict:
						# This repo has two configuration files, one for v4.0.0 and above with a different hidden size.
						lowercase         = "hf-internal-testing/test-two-configs"

						import transformers as new_transformers

						lowercase         = "v4.0.0"
						lowercase   ,							lowercase         = new_transformers.models.auto.AutoConfig.from_pretrained(
						    A__							,			return_unused_kwargs=A__				)
						self.assertEqual(new_configuration.hidden_size							,			2				)
						# This checks `_configuration_file` ia not kept in the kwargs by mistake.
						self.assertDictEqual(A__							,			{}				)

						# Testing an older version by monkey-patching the version in the module it's used.
						import transformers as old_transformers

						lowercase         = "v3.0.0"
						lowercase         = old_transformers.models.auto.AutoConfig.from_pretrained(A__				)
						self.assertEqual(old_configuration.hidden_size							,			7_6_8				)











'''simple docstring'''

# This is the module that test_patching.py uses to test patch_submodule()

import os  # noqa: this is just for tests
import os as renamed_os  # noqa: this is just for tests
from os import path  # noqa: this is just for tests
from os import path as renamed_path  # noqa: this is just for tests
from os.path import join  # noqa: this is just for tests
from os.path import join as renamed_join  # noqa: this is just for tests


__lowerCamelCase : Any											=			open  # noqa: we just need to have a builtin inside this module to test it properly




