
import json
import os
import sys
import tempfile
import unittest
from pathlib import Path
from shutil import copyfile

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

import transformers
from transformers import (
    CONFIG_MAPPING,
    FEATURE_EXTRACTOR_MAPPING,
    PROCESSOR_MAPPING,
    TOKENIZER_MAPPING,
    AutoConfig,
    AutoFeatureExtractor,
    AutoProcessor,
    AutoTokenizer,
    BertTokenizer,
    ProcessorMixin,
    WavaVecaConfig,
    WavaVecaFeatureExtractor,
    WavaVecaProcessor,
)
from transformers.testing_utils import TOKEN, USER, get_tests_dir, is_staging_test
from transformers.tokenization_utils import TOKENIZER_CONFIG_FILE
from transformers.utils import FEATURE_EXTRACTOR_NAME, is_tokenizers_available


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

from test_module.custom_configuration import CustomConfig  # noqa E402
from test_module.custom_feature_extraction import CustomFeatureExtractor  # noqa E402
from test_module.custom_processing import CustomProcessor  # noqa E402
from test_module.custom_tokenization import CustomTokenizer  # noqa E402


__lowerCamelCase							=       get_tests_dir("""fixtures/dummy_feature_extractor_config.json""")
__lowerCamelCase							=       get_tests_dir("""fixtures/vocab.json""")
__lowerCamelCase							=       get_tests_dir("""fixtures""")






class   UpperCAmelCase		(						unittest.TestCase ):
   A__					:		List[str]						     =     ["[UNK]", "[CLS]", "[SEP]", "[PAD]", "[MASK]", "bla", "blou"]



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


          '''simple docstring'''


          snake_case       :   str							=      0



   def 				_SCREAMING_SNAKE_CASE   (self		:					Optional[int] )		->	Any:


          '''simple docstring'''


          snake_case       :   Optional[int]							=      AutoProcessor.from_pretrained("facebook/wav2vec2-base-960h" )
          self.assertIsInstance(snake_case__    ,						snake_case__ )



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


          '''simple docstring'''


          with tempfile.TemporaryDirectory() as tmpdirname:
                 snake_case       :   List[Any]							=      WavaVecaConfig()
                 snake_case       :   Any							=      AutoProcessor.from_pretrained("facebook/wav2vec2-base-960h" )

                 # save in new folder
                 model_config.save_pretrained(snake_case__ )
                 processor.save_pretrained(snake_case__ )

                 snake_case       :   Any							=      AutoProcessor.from_pretrained(snake_case__ )

          self.assertIsInstance(snake_case__    ,						snake_case__ )



   def 				_SCREAMING_SNAKE_CASE   (self		:					Optional[int] )		->	Optional[Any]:


          '''simple docstring'''


          with tempfile.TemporaryDirectory() as tmpdirname:
                 # copy relevant files
                 copyfile(snake_case__    ,						os.path.join(snake_case__    ,						snake_case__ ) )
                 copyfile(snake_case__    ,						os.path.join(snake_case__    ,						"vocab.json" ) )

                 snake_case       :   List[str]							=      AutoProcessor.from_pretrained(snake_case__ )

          self.assertIsInstance(snake_case__    ,						snake_case__ )



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


          '''simple docstring'''


          with tempfile.TemporaryDirectory() as tmpdirname:
                 snake_case       :   int							=      WavaVecaFeatureExtractor()
                 snake_case       :   Any							=      AutoTokenizer.from_pretrained("facebook/wav2vec2-base-960h" )

                 snake_case       :   List[str]							=      WavaVecaProcessor(snake_case__    ,						snake_case__ )

                 # save in new folder
                 processor.save_pretrained(snake_case__ )

                 # drop `processor_class` in tokenizer
                 with open(os.path.join(snake_case__    ,						snake_case__ )    ,						"r" ) as f:
                        snake_case       :   int							=      json.load(snake_case__ )
                        config_dict.pop("processor_class" )

                 with open(os.path.join(snake_case__    ,						snake_case__ )    ,						"w" ) as f:
                        f.write(json.dumps(snake_case__ ) )

                 snake_case       :   Tuple							=      AutoProcessor.from_pretrained(snake_case__ )

          self.assertIsInstance(snake_case__    ,						snake_case__ )



   def 				_SCREAMING_SNAKE_CASE   (self		:					Optional[int] )		->	Tuple:


          '''simple docstring'''


          with tempfile.TemporaryDirectory() as tmpdirname:
                 snake_case       :   int							=      WavaVecaFeatureExtractor()
                 snake_case       :   Tuple							=      AutoTokenizer.from_pretrained("facebook/wav2vec2-base-960h" )

                 snake_case       :   Dict							=      WavaVecaProcessor(snake_case__    ,						snake_case__ )

                 # save in new folder
                 processor.save_pretrained(snake_case__ )

                 # drop `processor_class` in feature extractor
                 with open(os.path.join(snake_case__    ,						snake_case__ )    ,						"r" ) as f:
                        snake_case       :   List[str]							=      json.load(snake_case__ )
                        config_dict.pop("processor_class" )

                 with open(os.path.join(snake_case__    ,						snake_case__ )    ,						"w" ) as f:
                        f.write(json.dumps(snake_case__ ) )

                 snake_case       :   Union[str, Any]							=      AutoProcessor.from_pretrained(snake_case__ )

          self.assertIsInstance(snake_case__    ,						snake_case__ )



   def 				_SCREAMING_SNAKE_CASE   (self		:					Dict )		->	Tuple:


          '''simple docstring'''


          with tempfile.TemporaryDirectory() as tmpdirname:
                 snake_case       :   Tuple							=      WavaVecaConfig(processor_class="Wav2Vec2Processor" )
                 model_config.save_pretrained(snake_case__ )
                 # copy relevant files
                 copyfile(snake_case__    ,						os.path.join(snake_case__    ,						"vocab.json" ) )
                 # create emtpy sample processor
                 with open(os.path.join(snake_case__    ,						snake_case__ )    ,						"w" ) as f:
                        f.write("{}" )

                 snake_case       :   Union[str, Any]							=      AutoProcessor.from_pretrained(snake_case__ )

          self.assertIsInstance(snake_case__    ,						snake_case__ )



   def 				_SCREAMING_SNAKE_CASE   (self		:					Tuple )		->	Optional[int]:


          '''simple docstring'''


          with self.assertRaises(snake_case__ ):
                 snake_case       :   Optional[int]							=      AutoProcessor.from_pretrained("hf-internal-testing/test_dynamic_processor" )
          # If remote code is disabled, we can't load this config.
          with self.assertRaises(snake_case__ ):
                 snake_case       :   List[Any]							=      AutoProcessor.from_pretrained(
                     "hf-internal-testing/test_dynamic_processor"    ,						trust_remote_code=snake_case__ )

          snake_case       :   Optional[int]							=      AutoProcessor.from_pretrained("hf-internal-testing/test_dynamic_processor"    ,						trust_remote_code=snake_case__ )
          self.assertTrue(processor.special_attribute_present )
          self.assertEqual(processor.__class__.__name__    ,						"NewProcessor" )

          snake_case       :   List[Any]							=      processor.feature_extractor
          self.assertTrue(feature_extractor.special_attribute_present )
          self.assertEqual(feature_extractor.__class__.__name__    ,						"NewFeatureExtractor" )

          snake_case       :   str							=      processor.tokenizer
          self.assertTrue(tokenizer.special_attribute_present )
          if is_tokenizers_available():
                 self.assertEqual(tokenizer.__class__.__name__    ,						"NewTokenizerFast" )

                 # Test we can also load the slow version
                 snake_case       :   int							=      AutoProcessor.from_pretrained(
                     "hf-internal-testing/test_dynamic_processor"    ,						trust_remote_code=snake_case__    ,						use_fast=snake_case__ )
                 snake_case       :   Dict							=      new_processor.tokenizer
                 self.assertTrue(new_tokenizer.special_attribute_present )
                 self.assertEqual(new_tokenizer.__class__.__name__    ,						"NewTokenizer" )
          else:
                 self.assertEqual(tokenizer.__class__.__name__    ,						"NewTokenizer" )



   def 				_SCREAMING_SNAKE_CASE   (self		:					Optional[int] )		->	Optional[Any]:


          '''simple docstring'''


          try:
                 AutoConfig.register("custom"    ,						snake_case__ )
                 AutoFeatureExtractor.register(snake_case__    ,						snake_case__ )
                 AutoTokenizer.register(snake_case__    ,						slow_tokenizer_class=snake_case__ )
                 AutoProcessor.register(snake_case__    ,						snake_case__ )
                 # Trying to register something existing in the Transformers library will raise an error
                 with self.assertRaises(snake_case__ ):
                        AutoProcessor.register(snake_case__    ,						snake_case__ )

                 # Now that the config is registered, it can be used as any other config with the auto-API
                 snake_case       :   List[Any]							=      CustomFeatureExtractor.from_pretrained(snake_case__ )

                 with tempfile.TemporaryDirectory() as tmp_dir:
                        snake_case       :   List[str]							=      os.path.join(snake_case__    ,						"vocab.txt" )
                        with open(snake_case__    ,						"w"    ,						encoding="utf-8" ) as vocab_writer:
                               vocab_writer.write("".join([x + "\n" for x in self.vocab_tokens] ) )
                        snake_case       :   Optional[int]							=      CustomTokenizer(snake_case__ )

                 snake_case       :   List[Any]							=      CustomProcessor(snake_case__    ,						snake_case__ )

                 with tempfile.TemporaryDirectory() as tmp_dir:
                        processor.save_pretrained(snake_case__ )
                        snake_case       :   str							=      AutoProcessor.from_pretrained(snake_case__ )
                        self.assertIsInstance(snake_case__    ,						snake_case__ )

          finally:
                 if "custom" in CONFIG_MAPPING._extra_content:
                        del CONFIG_MAPPING._extra_content["custom"]
                 if CustomConfig in FEATURE_EXTRACTOR_MAPPING._extra_content:
                        del FEATURE_EXTRACTOR_MAPPING._extra_content[CustomConfig]
                 if CustomConfig in TOKENIZER_MAPPING._extra_content:
                        del TOKENIZER_MAPPING._extra_content[CustomConfig]
                 if CustomConfig in PROCESSOR_MAPPING._extra_content:
                        del PROCESSOR_MAPPING._extra_content[CustomConfig]



   def 				_SCREAMING_SNAKE_CASE   (self		:					int )		->	Union[str, Any]:


          '''simple docstring'''


          class   UpperCAmelCase		(						A_ ):
             A__					:		Tuple						     =     False






          class   UpperCAmelCase		(						A_ ):
             A__					:		Tuple						     =     False






          class   UpperCAmelCase		(						A_ ):
             A__					:		Dict						     =     "AutoFeatureExtractor"
             A__					:		str						     =     "AutoTokenizer"
             A__					:		Dict						     =     False

          try:
                 AutoConfig.register("custom"    ,						snake_case__ )
                 AutoFeatureExtractor.register(snake_case__    ,						snake_case__ )
                 AutoTokenizer.register(snake_case__    ,						slow_tokenizer_class=snake_case__ )
                 AutoProcessor.register(snake_case__    ,						snake_case__ )
                 # If remote code is not set, the default is to use local classes.
                 snake_case       :   Optional[Any]							=      AutoProcessor.from_pretrained("hf-internal-testing/test_dynamic_processor" )
                 self.assertEqual(processor.__class__.__name__    ,						"NewProcessor" )
                 self.assertFalse(processor.special_attribute_present )
                 self.assertFalse(processor.feature_extractor.special_attribute_present )
                 self.assertFalse(processor.tokenizer.special_attribute_present )

                 # If remote code is disabled, we load the local ones.
                 snake_case       :   Optional[int]							=      AutoProcessor.from_pretrained(
                     "hf-internal-testing/test_dynamic_processor"    ,						trust_remote_code=snake_case__ )
                 self.assertEqual(processor.__class__.__name__    ,						"NewProcessor" )
                 self.assertFalse(processor.special_attribute_present )
                 self.assertFalse(processor.feature_extractor.special_attribute_present )
                 self.assertFalse(processor.tokenizer.special_attribute_present )

                 # If remote is enabled, we load from the Hub.
                 snake_case       :   Optional[int]							=      AutoProcessor.from_pretrained(
                     "hf-internal-testing/test_dynamic_processor"    ,						trust_remote_code=snake_case__ )
                 self.assertEqual(processor.__class__.__name__    ,						"NewProcessor" )
                 self.assertTrue(processor.special_attribute_present )
                 self.assertTrue(processor.feature_extractor.special_attribute_present )
                 self.assertTrue(processor.tokenizer.special_attribute_present )

          finally:
                 if "custom" in CONFIG_MAPPING._extra_content:
                        del CONFIG_MAPPING._extra_content["custom"]
                 if CustomConfig in FEATURE_EXTRACTOR_MAPPING._extra_content:
                        del FEATURE_EXTRACTOR_MAPPING._extra_content[CustomConfig]
                 if CustomConfig in TOKENIZER_MAPPING._extra_content:
                        del TOKENIZER_MAPPING._extra_content[CustomConfig]
                 if CustomConfig in PROCESSOR_MAPPING._extra_content:
                        del PROCESSOR_MAPPING._extra_content[CustomConfig]



   def 				_SCREAMING_SNAKE_CASE   (self		:					Dict )		->	Any:


          '''simple docstring'''


          snake_case       :   str							=      AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-bert" )
          self.assertEqual(processor.__class__.__name__    ,						"BertTokenizerFast" )



   def 				_SCREAMING_SNAKE_CASE   (self		:					Optional[int] )		->	Tuple:


          '''simple docstring'''


          snake_case       :   Dict							=      AutoProcessor.from_pretrained("hf-internal-testing/tiny-random-convnext" )
          self.assertEqual(processor.__class__.__name__    ,						"ConvNextImageProcessor" )






@is_staging_test
class   UpperCAmelCase		(						unittest.TestCase ):
   A__					:		Dict						     =     ["[UNK]", "[CLS]", "[SEP]", "[PAD]", "[MASK]", "bla", "blou"]



   @classmethod
   def 				_SCREAMING_SNAKE_CASE   (cls		:					Dict )		->	List[Any]:


          '''simple docstring'''


          snake_case       :   Any							=      TOKEN
          HfFolder.save_token(snake_case__ )



   @classmethod
   def 				_SCREAMING_SNAKE_CASE   (cls		:					Optional[Any] )		->	Tuple:


          '''simple docstring'''


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

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

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



   def 				_SCREAMING_SNAKE_CASE   (self		:					Union[str, Any] )		->	Tuple:


          '''simple docstring'''


          snake_case       :   Dict							=      WavaVecaProcessor.from_pretrained(snake_case__ )
          with tempfile.TemporaryDirectory() as tmp_dir:
                 processor.save_pretrained(
                     os.path.join(snake_case__    ,						"test-processor" )    ,						push_to_hub=snake_case__    ,						use_auth_token=self._token )

                 snake_case       :   Tuple							=      WavaVecaProcessor.from_pretrained(f"""{USER}/test-processor""" )
                 for k, v in processor.feature_extractor.__dict__.items():
                        self.assertEqual(snake_case__    ,						getattr(new_processor.feature_extractor    ,						snake_case__ ) )
                 self.assertDictEqual(new_processor.tokenizer.get_vocab()    ,						processor.tokenizer.get_vocab() )



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


          '''simple docstring'''


          snake_case       :   int							=      WavaVecaProcessor.from_pretrained(snake_case__ )

          with tempfile.TemporaryDirectory() as tmp_dir:
                 processor.save_pretrained(
                     os.path.join(snake_case__    ,						"test-processor-org" )    ,						push_to_hub=snake_case__    ,						use_auth_token=self._token    ,						organization="valid_org"    ,						)

                 snake_case       :   Dict							=      WavaVecaProcessor.from_pretrained("valid_org/test-processor-org" )
                 for k, v in processor.feature_extractor.__dict__.items():
                        self.assertEqual(snake_case__    ,						getattr(new_processor.feature_extractor    ,						snake_case__ ) )
                 self.assertDictEqual(new_processor.tokenizer.get_vocab()    ,						processor.tokenizer.get_vocab() )



   def 				_SCREAMING_SNAKE_CASE   (self		:					int )		->	int:


          '''simple docstring'''


          CustomFeatureExtractor.register_for_auto_class()
          CustomTokenizer.register_for_auto_class()
          CustomProcessor.register_for_auto_class()

          snake_case       :   List[Any]							=      CustomFeatureExtractor.from_pretrained(snake_case__ )

          with tempfile.TemporaryDirectory() as tmp_dir:
                 snake_case       :   Any							=      os.path.join(snake_case__    ,						"vocab.txt" )
                 with open(snake_case__    ,						"w"    ,						encoding="utf-8" ) as vocab_writer:
                        vocab_writer.write("".join([x + "\n" for x in self.vocab_tokens] ) )
                 snake_case       :   str							=      CustomTokenizer(snake_case__ )

          snake_case       :   Tuple							=      CustomProcessor(snake_case__    ,						snake_case__ )

          with tempfile.TemporaryDirectory() as tmp_dir:
                 create_repo(f"""{USER}/test-dynamic-processor"""    ,						token=self._token )
                 snake_case       :   Optional[Any]							=      Repository(snake_case__    ,						clone_from=f"""{USER}/test-dynamic-processor"""    ,						token=self._token )
                 processor.save_pretrained(snake_case__ )

                 # This has added the proper auto_map field to the feature extractor config
                 self.assertDictEqual(
                     processor.feature_extractor.auto_map    ,						{
                         "AutoFeatureExtractor": "custom_feature_extraction.CustomFeatureExtractor",
                         "AutoProcessor": "custom_processing.CustomProcessor",
                     }    ,						)

                 # This has added the proper auto_map field to the tokenizer config
                 with open(os.path.join(snake_case__    ,						"tokenizer_config.json" ) ) as f:
                        snake_case       :   List[str]							=      json.load(snake_case__ )
                 self.assertDictEqual(
                     tokenizer_config["auto_map"]    ,						{
                         "AutoTokenizer": ["custom_tokenization.CustomTokenizer", None],
                         "AutoProcessor": "custom_processing.CustomProcessor",
                     }    ,						)

                 # The code has been copied from fixtures
                 self.assertTrue(os.path.isfile(os.path.join(snake_case__    ,						"custom_feature_extraction.py" ) ) )
                 self.assertTrue(os.path.isfile(os.path.join(snake_case__    ,						"custom_tokenization.py" ) ) )
                 self.assertTrue(os.path.isfile(os.path.join(snake_case__    ,						"custom_processing.py" ) ) )

                 repo.push_to_hub()

          snake_case       :   int							=      AutoProcessor.from_pretrained(f"""{USER}/test-dynamic-processor"""    ,						trust_remote_code=snake_case__ )
          # Can't make an isinstance check because the new_processor is from the CustomProcessor class of a dynamic module
          self.assertEqual(new_processor.__class__.__name__    ,						"CustomProcessor" )







__lowerCamelCase							=       {
    "joule": 1.0,
    "kilojoule": 10_00,
    "megajoule": 1_00_00_00,
    "gigajoule": 10_00_00_00_00,
    "wattsecond": 1.0,
    "watthour": 36_00,
    "kilowatthour": 3_60_00_00,
    "newtonmeter": 1.0,
    "calorie_nutr": 41_86.8,
    "kilocalorie_nutr": 4_18_68_00.00,
    "electronvolt": 1.602_176_634e-19,
    "britishthermalunit_it": 10_55.0_55_85,
    "footpound": 1.35_5818,
}


def 			UpperCamelCase					(    __lowerCamelCase     : str				,					__lowerCamelCase     : str				,					__lowerCamelCase     : float    ):
       if to_type not in ENERGY_CONVERSION or from_type not in ENERGY_CONVERSION:
              snake_case       :   List[Any]							=      (
                  f"""Incorrect 'from_type' or 'to_type' value: {from_type!r}, {to_type!r}\n"""
                  f"""Valid values are: {', '.join(__lowerCamelCase    )}"""
              )
              raise ValueError(__lowerCamelCase    )
       return value * ENERGY_CONVERSION[from_type] / ENERGY_CONVERSION[to_type]


if __name__ == "__main__":
   import doctest

   doctest.testmod()





