





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


_UpperCAmelCase             =							{
    '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        snake_case_    (		unittest.TestCase						):






  @classmethod
  def 				UpperCAmelCase__						(							cls			:					Optional[int]      )->Union[str, Any]:






   '''simple docstring'''

   __lowerCAmelCase					:	Any      =       TOKEN
   HfFolder.save_token(_snake_case      )






  @classmethod
  def 				UpperCAmelCase__						(							cls			:					int      )->Dict:






   '''simple docstring'''

   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			:					Optional[Any]      )->Union[str, Any]:






   '''simple docstring'''

   __lowerCAmelCase					:	Any      =       BertConfig(
       vocab_size=99       ,  hidden_size=32       ,  num_hidden_layers=5       ,  num_attention_heads=4       ,  intermediate_size=37      )
   config.push_to_hub("""test-config"""       ,  use_auth_token=self._token      )

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

        # 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(_snake_case       ,  repo_id="""test-config"""       ,  push_to_hub=_snake_case       ,  use_auth_token=self._token      )

   __lowerCAmelCase					:	Optional[int]      =       BertConfig.from_pretrained(F'''{USER}/test-config'''      )
   for k, v in config.to_dict().items():
    if k != "transformers_version":
     self.assertEqual(_snake_case       ,  getattr(_snake_case       ,  _snake_case      )      )






  def 				UpperCAmelCase__						(							self			:					Optional[int]      )->List[str]:






   '''simple docstring'''

   __lowerCAmelCase					:	List[str]      =       BertConfig(
       vocab_size=99       ,  hidden_size=32       ,  num_hidden_layers=5       ,  num_attention_heads=4       ,  intermediate_size=37      )
   config.push_to_hub("""valid_org/test-config-org"""       ,  use_auth_token=self._token      )

   __lowerCAmelCase					:	List[Any]      =       BertConfig.from_pretrained("""valid_org/test-config-org"""      )
   for k, v in config.to_dict().items():
    if k != "transformers_version":
     self.assertEqual(_snake_case       ,  getattr(_snake_case       ,  _snake_case      )      )

        # 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(
        _snake_case       ,  repo_id="""valid_org/test-config-org"""       ,  push_to_hub=_snake_case       ,  use_auth_token=self._token      )

   __lowerCAmelCase					:	Optional[Any]      =       BertConfig.from_pretrained("""valid_org/test-config-org"""      )
   for k, v in config.to_dict().items():
    if k != "transformers_version":
     self.assertEqual(_snake_case       ,  getattr(_snake_case       ,  _snake_case      )      )






  def 				UpperCAmelCase__						(							self			:					Dict      )->Dict:






   '''simple docstring'''

   CustomConfig.register_for_auto_class()
   __lowerCAmelCase					:	Dict      =       CustomConfig(attribute=42      )

   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"""}      )

   __lowerCAmelCase					:	str      =       AutoConfig.from_pretrained(F'''{USER}/test-dynamic-config'''       ,  trust_remote_code=_snake_case      )
   # 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       ,  42      )






class        snake_case_    (		unittest.TestCase						):






  def 				UpperCAmelCase__						(							self			:					List[Any]      )->Union[str, Any]:






   '''simple docstring'''

   __lowerCAmelCase					:	Any      =       GPTaConfig()

   # attempt to modify each of int/float/bool/str config records and verify they were updated
   __lowerCAmelCase					:	List[str]      =       c.n_embd + 1  # int
   __lowerCAmelCase					:	Union[str, Any]      =       c.resid_pdrop + 1.0  # float
   __lowerCAmelCase					:	Dict      =       not c.scale_attn_weights  # bool
   __lowerCAmelCase					:	Optional[Any]      =       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(_snake_case       ,  c.n_embd       ,  """mismatch for key: n_embd"""      )
   self.assertEqual(_snake_case       ,  c.resid_pdrop       ,  """mismatch for key: resid_pdrop"""      )
   self.assertEqual(_snake_case       ,  c.scale_attn_weights       ,  """mismatch for key: scale_attn_weights"""      )
   self.assertEqual(_snake_case       ,  c.summary_type       ,  """mismatch for key: summary_type"""      )






  def 				UpperCAmelCase__						(							self			:					List[Any]      )->Optional[Any]:






   '''simple docstring'''

   __lowerCAmelCase					:	Optional[Any]      =       PretrainedConfig()
   __lowerCAmelCase					:	List[Any]      =       [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(
       _snake_case       ,  ["""is_encoder_decoder""", """_name_or_path""", """_commit_hash""", """transformers_version"""]      )
   __lowerCAmelCase					:	Tuple      =       [key for key, value in config_common_kwargs.items() if value == getattr(_snake_case       ,  _snake_case      )]
   if len(_snake_case      ) > 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(_snake_case      )}.'''      )






  def 				UpperCAmelCase__						(							self			:					Optional[Any]      )->List[Any]:






   '''simple docstring'''

   with self.assertRaises(_snake_case      ):
    # config is in subfolder, the following should not work without specifying the subfolder
    __lowerCAmelCase					:	Optional[int]      =       BertConfig.from_pretrained("""hf-internal-testing/tiny-random-bert-subfolder"""      )

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

   self.assertIsNotNone(_snake_case      )






  def 				UpperCAmelCase__						(							self			:					Optional[int]      )->Optional[int]:






   '''simple docstring'''

   __lowerCAmelCase					:	int      =       mock.Mock()
   __lowerCAmelCase					:	List[Any]      =       500
   __lowerCAmelCase					:	List[str]      =       {}
   __lowerCAmelCase					:	Tuple      =       HTTPError
   __lowerCAmelCase					:	Optional[int]      =       {}

   # Download this model to make sure it's in the cache.
   __lowerCAmelCase					:	int      =       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=_snake_case      ) as mock_head:
    __lowerCAmelCase					:	Optional[Any]      =       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      )->Union[str, Any]:






   '''simple docstring'''

   __lowerCAmelCase					:	List[str]      =       BertConfig.from_pretrained(
       """https://huggingface.co/hf-internal-testing/tiny-random-bert/resolve/main/config.json"""      )






  def 				UpperCAmelCase__						(							self			:					List[Any]      )->Dict:






   '''simple docstring'''

   __lowerCAmelCase					:	Union[str, Any]      =       AutoConfig.from_pretrained("""bert-base-cased"""      )
   __lowerCAmelCase					:	str      =       ["""config.4.0.0.json"""]

   with tempfile.TemporaryDirectory() as tmp_dir:
    configuration.save_pretrained(_snake_case      )
    __lowerCAmelCase					:	Optional[Any]      =       2
    json.dump(configuration.to_dict()       ,  open(os.path.join(_snake_case       ,  """config.4.0.0.json"""      )       ,  """w"""      )      )

    # This should pick the new configuration file as the version of Transformers is > 4.0.0
    __lowerCAmelCase					:	Union[str, Any]      =       AutoConfig.from_pretrained(_snake_case      )
    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
    __lowerCAmelCase					:	str      =       ["""config.42.0.0.json"""]
    __lowerCAmelCase					:	Tuple      =       768
    configuration.save_pretrained(_snake_case      )
    shutil.move(os.path.join(_snake_case       ,  """config.4.0.0.json"""      )       ,  os.path.join(_snake_case       ,  """config.42.0.0.json"""      )      )
    __lowerCAmelCase					:	Any      =       AutoConfig.from_pretrained(_snake_case      )
    self.assertEqual(new_configuration.hidden_size       ,  768      )






  def 				UpperCAmelCase__						(							self			:					Any      )->Any:






   '''simple docstring'''

   __lowerCAmelCase					:	Optional[int]      =       """hf-internal-testing/test-two-configs"""

   import transformers as new_transformers

   __lowerCAmelCase					:	Optional[int]      =       """v4.0.0"""
   __lowerCAmelCase ,					__lowerCAmelCase					:	Dict      =       new_transformers.models.auto.AutoConfig.from_pretrained(
       _snake_case       ,  return_unused_kwargs=_snake_case      )
   self.assertEqual(new_configuration.hidden_size       ,  2      )
   # This checks `_configuration_file` ia not kept in the kwargs by mistake.
   self.assertDictEqual(_snake_case       ,  {}      )

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

   __lowerCAmelCase					:	Union[str, Any]      =       """v3.0.0"""
   __lowerCAmelCase					:	Union[str, Any]      =       old_transformers.models.auto.AutoConfig.from_pretrained(_snake_case      )
   self.assertEqual(old_configuration.hidden_size       ,  768      )






def 	_SCREAMING_SNAKE_CASE		(      SCREAMING_SNAKE_CASE		:list[int]   )      ->   int:
 if not numbers:
  return 0

 if not isinstance(SCREAMING_SNAKE_CASE   ,   (list, tuple)   ) or not all(
     isinstance(SCREAMING_SNAKE_CASE   ,   SCREAMING_SNAKE_CASE   ) for number in numbers   ):
  raise ValueError("""numbers must be an iterable of integers"""   )

 __lowerCAmelCase					:	int      =       numbers[0]

 for i in range(1   ,   len(SCREAMING_SNAKE_CASE   )   ):
  # update the maximum and minimum subarray products
  __lowerCAmelCase					:	List[str]      =       numbers[i]
  if number < 0:
   __lowerCAmelCase ,					__lowerCAmelCase					:	Union[str, Any]      =       min_till_now, max_till_now
  __lowerCAmelCase					:	Optional[int]      =       max(SCREAMING_SNAKE_CASE   ,   max_till_now * number   )
  __lowerCAmelCase					:	List[Any]      =       min(SCREAMING_SNAKE_CASE   ,   min_till_now * number   )

  # update the maximum product found till now
  __lowerCAmelCase					:	List[str]      =       max(SCREAMING_SNAKE_CASE   ,   SCREAMING_SNAKE_CASE   )

 return max_prod