

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__  :      int			=    {
    "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 						lowerCAmelCase__     (					cls):
      SCREAMING_SNAKE_CASE_									=						TOKEN
      HfFolder.save_token(_A)

  @classmethod
  def 						lowerCAmelCase__     (					cls):
      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 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						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)

      SCREAMING_SNAKE_CASE_									=						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)

      SCREAMING_SNAKE_CASE_									=						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 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						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)

      SCREAMING_SNAKE_CASE_									=						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)

      SCREAMING_SNAKE_CASE_									=						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 						lowerCAmelCase__     (					self):
      CustomConfig.register_for_auto_class()
      SCREAMING_SNAKE_CASE_									=						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'})

      SCREAMING_SNAKE_CASE_									=						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						,		42)






class       __snake_case  (    unittest.TestCase       ):

  def 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						GPTaConfig()

      # attempt to modify each of int/float/bool/str config records and verify they were updated
      SCREAMING_SNAKE_CASE_									=						c.n_embd + 1  # int
      SCREAMING_SNAKE_CASE_									=						c.resid_pdrop + 1.0  # float
      SCREAMING_SNAKE_CASE_									=						not c.scale_attn_weights  # bool
      SCREAMING_SNAKE_CASE_									=						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 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						PretrainedConfig()
      SCREAMING_SNAKE_CASE_									=						[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'])
      SCREAMING_SNAKE_CASE_									=						[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 						lowerCAmelCase__     (					self):
      with self.assertRaises(_A):
          # config is in subfolder, the following should not work without specifying the subfolder
          SCREAMING_SNAKE_CASE_									=						BertConfig.from_pretrained('hf-internal-testing/tiny-random-bert-subfolder')

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

      self.assertIsNotNone(_A)

  def 						lowerCAmelCase__     (					self):
      # A mock response for an HTTP head request to emulate server down
      SCREAMING_SNAKE_CASE_									=						mock.Mock()
      SCREAMING_SNAKE_CASE_									=						500
      SCREAMING_SNAKE_CASE_									=						{}
      SCREAMING_SNAKE_CASE_									=						HTTPError
      SCREAMING_SNAKE_CASE_									=						{}

      # Download this model to make sure it's in the cache.
      SCREAMING_SNAKE_CASE_									=						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:
          SCREAMING_SNAKE_CASE_									=						BertConfig.from_pretrained('hf-internal-testing/tiny-random-bert')
          # This check we did call the fake head request
          mock_head.assert_called()

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

  def 						lowerCAmelCase__     (					self):
      SCREAMING_SNAKE_CASE_									=						AutoConfig.from_pretrained('bert-base-cased')
      SCREAMING_SNAKE_CASE_									=						['config.4.0.0.json']

      with tempfile.TemporaryDirectory() as tmp_dir:
          configuration.save_pretrained(_A)
          SCREAMING_SNAKE_CASE_									=						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
          SCREAMING_SNAKE_CASE_									=						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
          SCREAMING_SNAKE_CASE_									=						['config.42.0.0.json']
          SCREAMING_SNAKE_CASE_									=						768
          configuration.save_pretrained(_A)
          shutil.move(os.path.join(_A						,		'config.4.0.0.json')						,		os.path.join(_A						,		'config.42.0.0.json'))
          SCREAMING_SNAKE_CASE_									=						AutoConfig.from_pretrained(_A)
          self.assertEqual(new_configuration.hidden_size						,		768)


  def 						lowerCAmelCase__     (					self):
      # This repo has two configuration files, one for v4.0.0 and above with a different hidden size.
      SCREAMING_SNAKE_CASE_									=						'hf-internal-testing/test-two-configs'

      import transformers as new_transformers

      SCREAMING_SNAKE_CASE_									=						'v4.0.0'
      SCREAMING_SNAKE_CASE_							,			SCREAMING_SNAKE_CASE_									=						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

      SCREAMING_SNAKE_CASE_									=						'v3.0.0'
      SCREAMING_SNAKE_CASE_									=						old_transformers.models.auto.AutoConfig.from_pretrained(_A)
      self.assertEqual(old_configuration.hidden_size						,		768)





import os
import zipfile

import requests
from get_ci_error_statistics import download_artifact, get_artifacts_links


def 					_UpperCAmelCase       (  _SCREAMING_SNAKE_CASE					:		List[Any] ,   _SCREAMING_SNAKE_CASE					:		int=7       ):





    """simple docstring"""


    SCREAMING_SNAKE_CASE_									=						None
    if token is not None:
        SCREAMING_SNAKE_CASE_									=						{'Accept': 'application/vnd.github+json', 'Authorization': f"""Bearer {token}"""}

    # The id of a workflow (not of a workflow run)
    SCREAMING_SNAKE_CASE_									=						'636036'

    SCREAMING_SNAKE_CASE_									=						f"""https://api.github.com/repos/huggingface/transformers/actions/workflows/{workflow_id}/runs"""
    # On `main` branch + event being `schedule` + not returning PRs + only `num_runs` results
    url += f"""?branch=main&event=schedule&exclude_pull_requests=true&per_page={num_runs}"""

    SCREAMING_SNAKE_CASE_									=						requests.get(_SCREAMING_SNAKE_CASE ,   headers=_SCREAMING_SNAKE_CASE       ).json()

    return result["workflow_runs"]


def 					_UpperCAmelCase       (  _SCREAMING_SNAKE_CASE					:		str       ):





    """simple docstring"""


    SCREAMING_SNAKE_CASE_									=						get_daily_ci_runs(_SCREAMING_SNAKE_CASE       )
    SCREAMING_SNAKE_CASE_									=						None
    for workflow_run in workflow_runs:
        if workflow_run["status"] == "completed":
            SCREAMING_SNAKE_CASE_									=						workflow_run['id']
            break

    return workflow_run_id


def 					_UpperCAmelCase       (  _SCREAMING_SNAKE_CASE					:		int ,   _SCREAMING_SNAKE_CASE					:		str ,   _SCREAMING_SNAKE_CASE					:		Optional[int]       ):





    """simple docstring"""


    SCREAMING_SNAKE_CASE_									=						get_last_daily_ci_runs(_SCREAMING_SNAKE_CASE       )
    if workflow_run_id is not None:
        SCREAMING_SNAKE_CASE_									=						get_artifacts_links(worflow_run_id=_SCREAMING_SNAKE_CASE ,   token=_SCREAMING_SNAKE_CASE       )
        for artifact_name in artifact_names:
            if artifact_name in artifacts_links:
                SCREAMING_SNAKE_CASE_									=						artifacts_links[artifact_name]
                download_artifact(
                    artifact_name=_SCREAMING_SNAKE_CASE ,   artifact_url=_SCREAMING_SNAKE_CASE ,   output_dir=_SCREAMING_SNAKE_CASE ,   token=_SCREAMING_SNAKE_CASE       )





def 					_UpperCAmelCase       (  _SCREAMING_SNAKE_CASE					:		List[Any] ,   _SCREAMING_SNAKE_CASE					:		str ,   _SCREAMING_SNAKE_CASE					:		List[Any]       ):





    """simple docstring"""


    get_last_daily_ci_artifacts(_SCREAMING_SNAKE_CASE ,   _SCREAMING_SNAKE_CASE ,   _SCREAMING_SNAKE_CASE       )

    SCREAMING_SNAKE_CASE_									=						{}
    for artifact_name in artifact_names:
        SCREAMING_SNAKE_CASE_									=						os.path.join(_SCREAMING_SNAKE_CASE ,   f"""{artifact_name}.zip"""       )
        if os.path.isfile(_SCREAMING_SNAKE_CASE       ):
            SCREAMING_SNAKE_CASE_									=						{}
            with zipfile.ZipFile(_SCREAMING_SNAKE_CASE       ) as z:
                for filename in z.namelist():
                    if not os.path.isdir(_SCREAMING_SNAKE_CASE       ):
                        # read the file
                        with z.open(_SCREAMING_SNAKE_CASE       ) as f:
                            SCREAMING_SNAKE_CASE_									=						f.read().decode('UTF-8'       )

    return results


