


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


SCREAMING_SNAKE_CASE       :		int         =				"src/transformers"


# Matches is_xxx_available()
SCREAMING_SNAKE_CASE       :		Optional[Any]         =				re.compile(r"is\_([a-z_]*)_available()")
# Catches a one-line _import_struct = {xxx}
SCREAMING_SNAKE_CASE       :		Any         =				re.compile(r"^_import_structure\s+=\s+\{([^\}]+)\}")
# Catches a line with a key-values pattern: "bla": ["foo", "bar"]
SCREAMING_SNAKE_CASE       :		str         =				re.compile(r"\s+\"\S*\":\s+\[([^\]]*)\]")
# Catches a line if not is_foo_available
SCREAMING_SNAKE_CASE       :		int         =				re.compile(r"^\s*if\s+not\s+is\_[a-z_]*\_available\(\)")
# Catches a line _import_struct["bla"].append("foo")
SCREAMING_SNAKE_CASE       :		Any         =				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       :		str         =				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       :		Dict         =				re.compile("^\s+\"([^\"]+)\",")
# Catches a line with objects between brackets only:    ["foo", "bar"],
SCREAMING_SNAKE_CASE       :		Optional[int]         =				re.compile("^\s+\[([^\]]+)\]")
# Catches a line with from foo import bar, bla, boo
SCREAMING_SNAKE_CASE       :		List[str]         =				re.compile(r"\s+from\s+\S*\s+import\s+([^\(\s].*)\n")
# Catches a line with try:
SCREAMING_SNAKE_CASE       :		int         =				re.compile(r"^\s*try:")
# Catches a line with else:
SCREAMING_SNAKE_CASE       :		str         =				re.compile(r"^\s*else:")




def     UpperCamelCase    (      _a		)					->							Tuple:
     '''simple docstring'''





     if _re_test_backend.search(_a		) is None:
          return None
     lowercase_       :List[str]     =       [b[0] for b in _re_backend.findall(_a		)]
     backends.sort()
     return "_and_".join(_a		)




def     UpperCamelCase    (      _a		)					->							Any:
     '''simple docstring'''





     with open(_a       ,		'''r'''       ,		encoding='''utf-8'''       ,		newline='''\n'''		) as f:
          lowercase_       :List[str]     =       f.readlines()

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

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

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

     lowercase_       :Optional[int]     =       {'''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.
          lowercase_       :int     =       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:
               lowercase_       :Optional[Any]     =       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

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

               lowercase_       :Union[str, Any]     =       objects
          else:
               line_index += 1

    # At this stage we are in the TYPE_CHECKING part, first grab the objects without a specific backend
     lowercase_       :List[Any]     =       []
     while (
         line_index < len(_a		)
         and find_backend(lines[line_index]		) is None
         and not lines[line_index].startswith('''else'''		)
     ):
          lowercase_       :Dict     =       lines[line_index]
          lowercase_       :Any     =       _re_import.search(_a		)
          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

     lowercase_       :Dict     =       {'''none''': objects}
     # Let's continue with backend-specific objects
     while line_index < len(_a		):
          # If the line is an if is_backend_available, we grab all objects associated.
          lowercase_       :Union[str, Any]     =       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:
               lowercase_       :Tuple     =       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

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

               lowercase_       :Optional[int]     =       objects
          else:
               line_index += 1

     return import_dict_objects, type_hint_objects




def     UpperCamelCase    (      _a       ,		_a		)					->							Tuple:
     '''simple docstring'''





     def find_duplicates(_a		):
          return [k for k, v in collections.Counter(_a		).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!"]

     lowercase_       :Dict     =       []
     for key in import_dict_objects.keys():
          lowercase_       :Tuple     =       find_duplicates(import_dict_objects[key]		)
          if duplicate_imports:
               errors.append(f"Duplicate _import_structure definitions for: {duplicate_imports}"		)
          lowercase_       :Any     =       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]		)		):
               lowercase_       :Optional[Any]     =       '''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    (      )					->							Optional[Any]:
     '''simple docstring'''





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




def     UpperCamelCase    (      )					->							Optional[Any]:
     '''simple docstring'''





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


SCREAMING_SNAKE_CASE       :		Optional[int]         =				[
    "convert_pytorch_checkpoint_to_tf2",
    "modeling_flax_pytorch_utils",
]




def     UpperCamelCase    (      )					->							Optional[int]:
     '''simple docstring'''





     lowercase_       :Optional[Any]     =       importlib.util.spec_from_file_location(
         '''transformers'''       ,		os.path.join(_a       ,		'''__init__.py'''		)       ,		submodule_search_locations=[PATH_TO_TRANSFORMERS]       ,		)
     lowercase_       :Tuple     =       spec.loader.load_module()

     lowercase_       :str     =       [
         module
         for module in get_transformers_submodules()
         if module not in IGNORE_SUBMODULES and module not in transformers._import_structure.keys()
     ]
     if len(_a		) > 0:
          lowercase_       :List[Any]     =       '''\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()





from torch import nn




def     UpperCamelCase    (      _a		)					->							str:
     '''simple docstring'''





     if act_fn in ["swish", "silu"]:
          return nn.SiLU()
     elif act_fn == "mish":
          return nn.Mish()
     elif act_fn == "gelu":
          return nn.GELU()
     else:
          raise ValueError(f"Unsupported activation function: {act_fn}"		)

