


import collections
import importlib.util
import os
import re
from pathlib import Path


_SCREAMING_SNAKE_CASE     :					Optional[Any]      =       "src/transformers"


# Matches is_xxx_available()
_SCREAMING_SNAKE_CASE     :					Optional[int]      =       re.compile(R"is\_([a-z_]*)_available()")
# Catches a one-line _import_struct = {xxx}
_SCREAMING_SNAKE_CASE     :					int      =       re.compile(R"^_import_structure\s+=\s+\{([^\}]+)\}")
# Catches a line with a key-values pattern: "bla": ["foo", "bar"]
_SCREAMING_SNAKE_CASE     :					Tuple      =       re.compile(R"\s+\"\S*\":\s+\[([^\]]*)\]")
# Catches a line if not is_foo_available
_SCREAMING_SNAKE_CASE     :					List[str]      =       re.compile(R"^\s*if\s+not\s+is\_[a-z_]*\_available\(\)")
# Catches a line _import_struct["bla"].append("foo")
_SCREAMING_SNAKE_CASE     :					Tuple      =       re.compile(R"^\s*_import_structure\[\"\S*\"\]\.append\(\"(\S*)\"\)")
# Catches a line _import_struct["bla"].extend(["foo", "bar"]) or _import_struct["bla"] = ["foo", "bar"]
_SCREAMING_SNAKE_CASE     :					Optional[Any]      =       re.compile(R"^\s*_import_structure\[\S*\](?:\.extend\(|\s*=\s+)\[([^\]]*)\]")
# Catches a line with an object between quotes and a comma:     "MyModel",
_SCREAMING_SNAKE_CASE     :					List[str]      =       re.compile("^\s+\"([^\"]+)\",")
# Catches a line with objects between brackets only:    ["foo", "bar"],
_SCREAMING_SNAKE_CASE     :					Tuple      =       re.compile("^\s+\[([^\]]+)\]")
# Catches a line with from foo import bar, bla, boo
_SCREAMING_SNAKE_CASE     :					str      =       re.compile(R"\s+from\s+\S*\s+import\s+([^\(\s].*)\n")
# Catches a line with try:
_SCREAMING_SNAKE_CASE     :					Optional[int]      =       re.compile(R"^\s*try:")
# Catches a line with else:
_SCREAMING_SNAKE_CASE     :					Optional[int]      =       re.compile(R"^\s*else:")
def      UpperCAmelCase__		(UpperCamelCase_							):



     """simple docstring"""






     if _re_test_backend.search(UpperCamelCase_							) is None:
          return None
     snake_case											=							[b[0] for b in _re_backend.findall(UpperCamelCase_							)]
     backends.sort()
     return "_and_".join(UpperCamelCase_							)
def      UpperCAmelCase__		(UpperCamelCase_							):



     """simple docstring"""






     with open(UpperCamelCase_				,'''r'''				,encoding='''utf-8'''				,newline='''\n'''							) as f:
          snake_case											=							f.readlines()

     snake_case											=							0
     while line_index < len(UpperCamelCase_							) and not lines[line_index].startswith('''_import_structure = {'''							):
          line_index += 1

     # If this is a traditional init, just return.
     if line_index >= len(UpperCamelCase_							):
          return None

     # First grab the objects without a specific backend in _import_structure
     snake_case											=							[]
     while not lines[line_index].startswith('''if TYPE_CHECKING'''							) and find_backend(lines[line_index]							) is None:
          snake_case											=							lines[line_index]
          # If we have everything on a single line, let's deal with it.
          if _re_one_line_import_struct.search(UpperCamelCase_							):
               snake_case											=							_re_one_line_import_struct.search(UpperCamelCase_							).groups()[0]
               snake_case											=							re.findall('''\[([^\]]+)\]'''				,UpperCamelCase_							)
               for imp in imports:
                    objects.extend([obj[1:-1] for obj in imp.split(''', '''							)]							)
               line_index += 1
               continue
          snake_case											=							_re_import_struct_key_value.search(UpperCamelCase_							)
          if single_line_import_search is not None:
               snake_case											=							[obj[1:-1] for obj in single_line_import_search.groups()[0].split(''', '''							) if len(UpperCamelCase_							) > 0]
               objects.extend(UpperCamelCase_							)
          elif line.startswith(''' ''' * 8 + '''"'''							):
               objects.append(line[9:-3]							)
          line_index += 1

     snake_case											=							{'''none''': objects}
     # Let's continue with backend-specific objects in _import_structure
     while not lines[line_index].startswith('''if TYPE_CHECKING'''							):
          # If the line is an if not is_backend_available, we grab all objects associated.
          snake_case											=							find_backend(lines[line_index]							)
          # Check if the backend declaration is inside a try block:
          if _re_try.search(lines[line_index - 1]							) is None:
               snake_case											=							None

          if backend is not None:
               line_index += 1

               # Scroll until we hit the else block of try-except-else
               while _re_else.search(lines[line_index]							) is None:
                    line_index += 1

               line_index += 1

               snake_case											=							[]
               # Until we unindent, add backend objects to the list
               while len(lines[line_index]							) <= 1 or lines[line_index].startswith(''' ''' * 4							):
                    snake_case											=							lines[line_index]
                    if _re_import_struct_add_one.search(UpperCamelCase_							) is not None:
                         objects.append(_re_import_struct_add_one.search(UpperCamelCase_							).groups()[0]							)
                    elif _re_import_struct_add_many.search(UpperCamelCase_							) is not None:
                         snake_case											=							_re_import_struct_add_many.search(UpperCamelCase_							).groups()[0].split(''', '''							)
                         snake_case											=							[obj[1:-1] for obj in imports if len(UpperCamelCase_							) > 0]
                         objects.extend(UpperCamelCase_							)
                    elif _re_between_brackets.search(UpperCamelCase_							) is not None:
                         snake_case											=							_re_between_brackets.search(UpperCamelCase_							).groups()[0].split(''', '''							)
                         snake_case											=							[obj[1:-1] for obj in imports if len(UpperCamelCase_							) > 0]
                         objects.extend(UpperCamelCase_							)
                    elif _re_quote_object.search(UpperCamelCase_							) is not None:
                         objects.append(_re_quote_object.search(UpperCamelCase_							).groups()[0]							)
                    elif line.startswith(''' ''' * 8 + '''"'''							):
                         objects.append(line[9:-3]							)
                    elif line.startswith(''' ''' * 12 + '''"'''							):
                         objects.append(line[13:-3]							)
                    line_index += 1

               snake_case											=							objects
          else:
               line_index += 1

    # At this stage we are in the TYPE_CHECKING part, first grab the objects without a specific backend
     snake_case											=							[]
     while (
         line_index < len(UpperCamelCase_							)
         and find_backend(lines[line_index]							) is None
         and not lines[line_index].startswith('''else'''							)
     ):
          snake_case											=							lines[line_index]
          snake_case											=							_re_import.search(UpperCamelCase_							)
          if single_line_import_search is not None:
               objects.extend(single_line_import_search.groups()[0].split(''', '''							)							)
          elif line.startswith(''' ''' * 8							):
               objects.append(line[8:-2]							)
          line_index += 1

     snake_case											=							{'''none''': objects}
     # Let's continue with backend-specific objects
     while line_index < len(UpperCamelCase_							):
          # If the line is an if is_backend_available, we grab all objects associated.
          snake_case											=							find_backend(lines[line_index]							)
          # Check if the backend declaration is inside a try block:
          if _re_try.search(lines[line_index - 1]							) is None:
               snake_case											=							None

          if backend is not None:
               line_index += 1

               # Scroll until we hit the else block of try-except-else
               while _re_else.search(lines[line_index]							) is None:
                    line_index += 1

               line_index += 1

               snake_case											=							[]
               # Until we unindent, add backend objects to the list
               while len(lines[line_index]							) <= 1 or lines[line_index].startswith(''' ''' * 8							):
                    snake_case											=							lines[line_index]
                    snake_case											=							_re_import.search(UpperCamelCase_							)
                    if single_line_import_search is not None:
                         objects.extend(single_line_import_search.groups()[0].split(''', '''							)							)
                    elif line.startswith(''' ''' * 12							):
                         objects.append(line[12:-2]							)
                    line_index += 1

               snake_case											=							objects
          else:
               line_index += 1

     return import_dict_objects, type_hint_objects
def      UpperCAmelCase__		(UpperCamelCase_				,UpperCamelCase_							):



     """simple docstring"""






     def find_duplicates(UpperCamelCase_							):
          return [k for k, v in collections.Counter(UpperCamelCase_							).items() if v > 1]

     if list(import_dict_objects.keys()							) != list(type_hint_objects.keys()							):
          return ["Both sides of the init do not have the same backends!"]

     snake_case											=							[]
     for key in import_dict_objects.keys():
          snake_case											=							find_duplicates(import_dict_objects[key]							)
          if duplicate_imports:
               errors.append(F'''Duplicate _import_structure definitions for: {duplicate_imports}'''							)
          snake_case											=							find_duplicates(type_hint_objects[key]							)
          if duplicate_type_hints:
               errors.append(F'''Duplicate TYPE_CHECKING objects for: {duplicate_type_hints}'''							)

          if sorted(set(import_dict_objects[key]							)							) != sorted(set(type_hint_objects[key]							)							):
               snake_case											=							'''base imports''' if key == '''none''' else F'''{key} backend'''
               errors.append(F'''Differences for {name}:'''							)
               for a in type_hint_objects[key]:
                    if a not in import_dict_objects[key]:
                         errors.append(F'''  {a} in TYPE_HINT but not in _import_structure.'''							)
               for a in import_dict_objects[key]:
                    if a not in type_hint_objects[key]:
                         errors.append(F'''  {a} in _import_structure but not in TYPE_HINT.'''							)
     return errors
def      UpperCAmelCase__		():



     """simple docstring"""






     snake_case											=							[]
     for root, _, files in os.walk(UpperCamelCase_							):
          if "__init__.py" in files:
               snake_case											=							os.path.join(UpperCamelCase_				,'''__init__.py'''							)
               snake_case											=							parse_init(UpperCamelCase_							)
               if objects is not None:
                    snake_case											=							analyze_results(*UpperCamelCase_							)
                    if len(UpperCamelCase_							) > 0:
                         snake_case											=							F'''Problem in {fname}, both halves do not define the same objects.\n{errors[0]}'''
                         failures.append('''\n'''.join(UpperCamelCase_							)							)
     if len(UpperCamelCase_							) > 0:
          raise ValueError('''\n\n'''.join(UpperCamelCase_							)							)
def      UpperCAmelCase__		():



     """simple docstring"""






     snake_case											=							[]
     for path, directories, files in os.walk(UpperCamelCase_							):
          for folder in directories:
               # Ignore private modules
               if folder.startswith('''_'''							):
                    directories.remove(UpperCamelCase_							)
                    continue
               # Ignore leftovers from branches (empty folders apart from pycache)
               if len(list((Path(UpperCamelCase_							) / folder).glob('''*.py'''							)							)							) == 0:
                    continue
               snake_case											=							str((Path(UpperCamelCase_							) / folder).relative_to(UpperCamelCase_							)							)
               snake_case											=							short_path.replace(os.path.sep				,'''.'''							)
               submodules.append(UpperCamelCase_							)
          for fname in files:
               if fname == "__init__.py":
                    continue
               snake_case											=							str((Path(UpperCamelCase_							) / fname).relative_to(UpperCamelCase_							)							)
               snake_case											=							short_path.replace('''.py'''				,''''''							).replace(os.path.sep				,'''.'''							)
               if len(submodule.split('''.'''							)							) == 1:
                    submodules.append(UpperCamelCase_							)
     return submodules


_SCREAMING_SNAKE_CASE     :					int      =       [
    "convert_pytorch_checkpoint_to_tf2",
    "modeling_flax_pytorch_utils",
]
def      UpperCAmelCase__		():



     """simple docstring"""






     snake_case											=							importlib.util.spec_from_file_location(
         '''transformers'''				,os.path.join(UpperCamelCase_				,'''__init__.py'''							)				,submodule_search_locations=[PATH_TO_TRANSFORMERS]				,)
     snake_case											=							spec.loader.load_module()

     snake_case											=							[
         module
         for module in get_transformers_submodules()
         if module not in IGNORE_SUBMODULES and module not in transformers._import_structure.keys()
     ]
     if len(UpperCamelCase_							) > 0:
          snake_case											=							'''\n'''.join(F'''- {module}''' for module in module_not_registered							)
          raise ValueError(
              '''The following submodules are not properly registered in the main init of Transformers:\n'''
              F'''{list_of_modules}\n'''
              '''Make sure they appear somewhere in the keys of `_import_structure` with an empty list as value.'''							)


if __name__ == "__main__":
 check_all_inits()
 check_submodules()








import shutil
import tempfile
import unittest

import numpy as np
import pytest

from transformers import is_speech_available, is_vision_available
from transformers.testing_utils import require_torch


if is_vision_available():
 from transformers import TvltImageProcessor

if is_speech_available():
 from transformers import TvltFeatureExtractor

from transformers import TvltProcessor

@require_torch
class     A__   (       unittest.TestCase      ):







    """simple docstring"""

    def 						a_     (			self       ):
         snake_case											=							'''ZinengTang/tvlt-base'''
         snake_case											=							tempfile.mkdtemp()
    def 						a_     (			self     ,   **__snake_case       ):
         return TvltImageProcessor.from_pretrained(self.checkpoint     ,   **__snake_case       )
    def 						a_     (			self     ,   **__snake_case       ):
         return TvltFeatureExtractor.from_pretrained(self.checkpoint     ,   **__snake_case       )
    def 						a_     (			self       ):
         shutil.rmtree(self.tmpdirname       )
    def 						a_     (			self       ):
         snake_case											=							self.get_image_processor()
         snake_case											=							self.get_feature_extractor()

         snake_case											=							TvltProcessor(image_processor=__snake_case     ,   feature_extractor=__snake_case       )
         processor.save_pretrained(self.tmpdirname       )
         snake_case											=							TvltProcessor.from_pretrained(self.tmpdirname       )

         self.assertIsInstance(processor.feature_extractor     ,   __snake_case       )
         self.assertIsInstance(processor.image_processor     ,   __snake_case       )
    def 						a_     (			self       ):
         snake_case											=							self.get_image_processor()
         snake_case											=							self.get_feature_extractor()

         snake_case											=							TvltProcessor(image_processor=__snake_case     ,   feature_extractor=__snake_case       )

         snake_case											=							np.ones([1_2_0_0_0]       )

         snake_case											=							feature_extractor(__snake_case     ,   return_tensors='''np'''       )
         snake_case											=							processor(audio=__snake_case     ,   return_tensors='''np'''       )

         for key in audio_dict.keys():
              self.assertAlmostEqual(audio_dict[key].sum()     ,   input_processor[key].sum()     ,   delta=1E-2       )
    def 						a_     (			self       ):
         snake_case											=							self.get_image_processor()
         snake_case											=							self.get_feature_extractor()

         snake_case											=							TvltProcessor(image_processor=__snake_case     ,   feature_extractor=__snake_case       )

         snake_case											=							np.ones([3, 2_2_4, 2_2_4]       )

         snake_case											=							image_processor(__snake_case     ,   return_tensors='''np'''       )
         snake_case											=							processor(images=__snake_case     ,   return_tensors='''np'''       )

         for key in image_dict.keys():
              self.assertAlmostEqual(image_dict[key].sum()     ,   input_processor[key].sum()     ,   delta=1E-2       )
    def 						a_     (			self       ):
         snake_case											=							self.get_image_processor()
         snake_case											=							self.get_feature_extractor()

         snake_case											=							TvltProcessor(image_processor=__snake_case     ,   feature_extractor=__snake_case       )

         snake_case											=							np.ones([1_2_0_0_0]       )
         snake_case											=							np.ones([3, 2_2_4, 2_2_4]       )

         snake_case											=							processor(audio=__snake_case     ,   images=__snake_case       )

         self.assertListEqual(list(inputs.keys()       )     ,   ['''audio_values''', '''audio_mask''', '''pixel_values''', '''pixel_mask''']       )

         # test if it raises when no input is passed
         with pytest.raises(__snake_case       ):
              processor()




    def 						a_     (			self       ):
         snake_case											=							self.get_image_processor()
         snake_case											=							self.get_feature_extractor()

         snake_case											=							TvltProcessor(image_processor=__snake_case     ,   feature_extractor=__snake_case       )

         self.assertListEqual(
             processor.model_input_names     ,   image_processor.model_input_names + feature_extractor.model_input_names     ,   msg='''`processor` and `image_processor`+`feature_extractor` model input names do not match'''     ,   )




