





import importlib
import json
import os
import sys
import tempfile
import unittest
from pathlib import Path

import transformers
import transformers.models.auto
from transformers.models.auto.configuration_auto import CONFIG_MAPPING, AutoConfig
from transformers.models.bert.configuration_bert import BertConfig
from transformers.models.roberta.configuration_roberta import RobertaConfig
from transformers.testing_utils import DUMMY_UNKNOWN_IDENTIFIER, get_tests_dir


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

from test_module.custom_configuration import CustomConfig  # noqa E402


a__: str							   =						get_tests_dir('fixtures/dummy-config.json')




class        SCREAMING_SNAKE_CASE__					(		unittest.TestCase ):



 def 							UpperCamelCase     (				self					):
        A__		   =		0



 def 							UpperCamelCase     (				self					):
        self.assertIsNotNone(transformers.models.auto.__spec__					)
        self.assertIsNotNone(importlib.util.find_spec('''transformers.models.auto'''					)					)



 def 							UpperCamelCase     (				self					):
        A__		   =		AutoConfig.from_pretrained('''bert-base-uncased'''					)
        self.assertIsInstance(__lowerCamelCase,__lowerCamelCase					)



 def 							UpperCamelCase     (				self					):
        A__		   =		AutoConfig.from_pretrained(__lowerCamelCase					)
        self.assertIsInstance(__lowerCamelCase,__lowerCamelCase					)



 def 							UpperCamelCase     (				self					):
        A__		   =		AutoConfig.from_pretrained(__lowerCamelCase					)
        self.assertIsInstance(__lowerCamelCase,__lowerCamelCase					)



 def 							UpperCamelCase     (				self					):
        A__		   =		AutoConfig.for_model('''roberta'''					)
        self.assertIsInstance(__lowerCamelCase,__lowerCamelCase					)



 def 							UpperCamelCase     (				self					):
        with tempfile.TemporaryDirectory() as tmp_dir:
               # This model name contains bert and roberta, but roberta ends up being picked.
               A__		   =		os.path.join(__lowerCamelCase,'''fake-roberta'''					)
               os.makedirs(__lowerCamelCase,exist_ok=__lowerCamelCase					)
               with open(os.path.join(__lowerCamelCase,'''config.json'''					),'''w'''					) as f:
                      f.write(json.dumps({}					)					)
               A__		   =		AutoConfig.from_pretrained(__lowerCamelCase					)
               self.assertEqual(type(__lowerCamelCase					),__lowerCamelCase					)



 def 							UpperCamelCase     (				self					):
        try:
               AutoConfig.register('''custom''',__lowerCamelCase					)
               # Wrong model type will raise an error
               with self.assertRaises(__lowerCamelCase					):
                      AutoConfig.register('''model''',__lowerCamelCase					)
               # Trying to register something existing in the Transformers library will raise an error
               with self.assertRaises(__lowerCamelCase					):
                      AutoConfig.register('''bert''',__lowerCamelCase					)

               # Now that the config is registered, it can be used as any other config with the auto-API
               A__		   =		CustomConfig()
               with tempfile.TemporaryDirectory() as tmp_dir:
                      config.save_pretrained(__lowerCamelCase					)
                      A__		   =		AutoConfig.from_pretrained(__lowerCamelCase					)
                      self.assertIsInstance(__lowerCamelCase,__lowerCamelCase					)

        finally:
               if "custom" in CONFIG_MAPPING._extra_content:
                      del CONFIG_MAPPING._extra_content["custom"]



 def 							UpperCamelCase     (				self					):
        with self.assertRaisesRegex(
            __lowerCamelCase,'''bert-base is not a local folder and is not a valid model identifier'''					):
               A__		   =		AutoConfig.from_pretrained('''bert-base'''					)



 def 							UpperCamelCase     (				self					):
        with self.assertRaisesRegex(
            __lowerCamelCase,r'''aaaaaa is not a valid git identifier \(branch name, tag name or commit id\)'''					):
               A__		   =		AutoConfig.from_pretrained(__lowerCamelCase,revision='''aaaaaa'''					)



 def 							UpperCamelCase     (				self					):
        with self.assertRaisesRegex(
            __lowerCamelCase,'''hf-internal-testing/no-config-test-repo does not appear to have a file named config.json.''',):
               A__		   =		AutoConfig.from_pretrained('''hf-internal-testing/no-config-test-repo'''					)



 def 							UpperCamelCase     (				self					):
        # If remote code is not set, we will time out when asking whether to load the model.
        with self.assertRaises(__lowerCamelCase					):
               A__		   =		AutoConfig.from_pretrained('''hf-internal-testing/test_dynamic_model'''					)
        # If remote code is disabled, we can't load this config.
        with self.assertRaises(__lowerCamelCase					):
               A__		   =		AutoConfig.from_pretrained('''hf-internal-testing/test_dynamic_model''',trust_remote_code=__lowerCamelCase					)

        A__		   =		AutoConfig.from_pretrained('''hf-internal-testing/test_dynamic_model''',trust_remote_code=__lowerCamelCase					)
        self.assertEqual(config.__class__.__name__,'''NewModelConfig'''					)

        # Test config can be reloaded.
        with tempfile.TemporaryDirectory() as tmp_dir:
               config.save_pretrained(__lowerCamelCase					)
               A__		   =		AutoConfig.from_pretrained(__lowerCamelCase,trust_remote_code=__lowerCamelCase					)
        self.assertEqual(reloaded_config.__class__.__name__,'''NewModelConfig'''					)




 def 							UpperCamelCase     (				self					):




        class        SCREAMING_SNAKE_CASE__					(		UpperCamelCase__ ):
         __SCREAMING_SNAKE_CASE							=      '''new-model'''

        try:
               AutoConfig.register('''new-model''',__lowerCamelCase					)
               # If remote code is not set, the default is to use local
               A__		   =		AutoConfig.from_pretrained('''hf-internal-testing/test_dynamic_model'''					)
               self.assertEqual(config.__class__.__name__,'''NewModelConfigLocal'''					)

               # If remote code is disabled, we load the local one.
               A__		   =		AutoConfig.from_pretrained('''hf-internal-testing/test_dynamic_model''',trust_remote_code=__lowerCamelCase					)
               self.assertEqual(config.__class__.__name__,'''NewModelConfigLocal'''					)

               # If remote is enabled, we load from the Hub
               A__		   =		AutoConfig.from_pretrained('''hf-internal-testing/test_dynamic_model''',trust_remote_code=__lowerCamelCase					)
               self.assertEqual(config.__class__.__name__,'''NewModelConfig'''					)

        finally:
               if "new-model" in CONFIG_MAPPING._extra_content:
                      del CONFIG_MAPPING._extra_content["new-model"]












from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_torch_available,
)


a__: str							   =						{
    'configuration_mega': ['MEGA_PRETRAINED_CONFIG_ARCHIVE_MAP', 'MegaConfig', 'MegaOnnxConfig'],
}

try:
       if not is_torch_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       a__: Union[str, Any]							   =						[
           'MEGA_PRETRAINED_MODEL_ARCHIVE_LIST',
           'MegaForCausalLM',
           'MegaForMaskedLM',
           'MegaForMultipleChoice',
           'MegaForQuestionAnswering',
           'MegaForSequenceClassification',
           'MegaForTokenClassification',
           'MegaModel',
           'MegaPreTrainedModel',
       ]

if TYPE_CHECKING:
       from .configuration_mega import MEGA_PRETRAINED_CONFIG_ARCHIVE_MAP, MegaConfig, MegaOnnxConfig

       try:
              if not is_torch_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .modeling_mega import (
                  MEGA_PRETRAINED_MODEL_ARCHIVE_LIST,
                  MegaForCausalLM,
                  MegaForMaskedLM,
                  MegaForMultipleChoice,
                  MegaForQuestionAnswering,
                  MegaForSequenceClassification,
                  MegaForTokenClassification,
                  MegaModel,
                  MegaPreTrainedModel,
              )

else:
       import sys

       a__: Union[str, Any]							   =						_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)





