



'''simple docstring'''






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


lowerCAmelCase: int							=						'src/transformers'


# Matches is_xxx_available()
lowerCAmelCase: List[Any]							=						re.compile(r'is\_([a-z_]*)_available()')
# Catches a one-line _import_struct = {xxx}
lowerCAmelCase: Optional[Any]							=						re.compile(r'^_import_structure\s+=\s+\{([^\}]+)\}')
# Catches a line with a key-values pattern: "bla": ["foo", "bar"]
lowerCAmelCase: str							=						re.compile(r'\s+"\S*":\s+\[([^\]]*)\]')
# Catches a line if not is_foo_available
lowerCAmelCase: List[Any]							=						re.compile(r'^\s*if\s+not\s+is\_[a-z_]*\_available\(\)')
# Catches a line _import_struct["bla"].append("foo")
lowerCAmelCase: 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"]
lowerCAmelCase: Dict							=						re.compile(r'^\s*_import_structure\[\S*\](?:\.extend\(|\s*=\s+)\[([^\]]*)\]')
# Catches a line with an object between quotes and a comma:     "MyModel",
lowerCAmelCase: Any							=						re.compile('^\s+"([^"]+)",')
# Catches a line with objects between brackets only:    ["foo", "bar"],
lowerCAmelCase: Dict							=						re.compile('^\s+\[([^\]]+)\]')
# Catches a line with from foo import bar, bla, boo
lowerCAmelCase: Optional[int]							=						re.compile(r'\s+from\s+\S*\s+import\s+([^\(\s].*)\n')
# Catches a line with try:
lowerCAmelCase: Optional[int]							=						re.compile(r'^\s*try:')
# Catches a line with else:
lowerCAmelCase: Optional[int]							=						re.compile(r'^\s*else:')
def 				lowerCamelCase__	(						_A    ):
  if _re_test_backend.search(_A    ) is None:
    return None
  a						: Dict					       =					[b[0] for b in _re_backend.findall(_A    )]
  backends.sort()
  return "_and_".join(_A    )
def 				lowerCamelCase__	(						_A    ):
  with open(_A			,      'r'			,      encoding='utf-8'			,      newline='\n'    ) as f:
    a						: Optional[int]					       =					f.readlines()

  a						: Tuple					       =					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
  a						: Optional[int]					       =					[]
  while not lines[line_index].startswith('if TYPE_CHECKING'    ) and find_backend(lines[line_index]    ) is None:
    a						: 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    ):
      a						: Any					       =					_re_one_line_import_struct.search(_A    ).groups()[0]
      a						: Union[str, Any]					       =					re.findall('\[([^\]]+)\]'			,      _A    )
      for imp in imports:
        objects.extend([obj[1:-1] for obj in imp.split(', '    )]    )
      line_index += 1
      continue
    a						: List[Any]					       =					_re_import_struct_key_value.search(_A    )
    if single_line_import_search is not None:
      a						: 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

  a						: List[str]					       =					{'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.
    a						: Dict					       =					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:
      a						: List[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

      a						: int					       =					[]
      # Until we unindent, add backend objects to the list
      while len(lines[line_index]    ) <= 1 or lines[line_index].startswith(' ' * 4    ):
        a						: Union[str, Any]					       =					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:
          a						: Tuple					       =					_re_import_struct_add_many.search(_A    ).groups()[0].split(', '    )
          a						: Any					       =					[obj[1:-1] for obj in imports if len(_A    ) > 0]
          objects.extend(_A    )
        elif _re_between_brackets.search(_A    ) is not None:
          a						: Dict					       =					_re_between_brackets.search(_A    ).groups()[0].split(', '    )
          a						: List[str]					       =					[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(' ' * 12 + '"'    ):
          objects.append(line[13:-3]    )
        line_index += 1

      a						: Dict					       =					objects
    else:
      line_index += 1

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

  a						: List[str]					       =					{'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.
    a						: Tuple					       =					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:
      a						: 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

      a						: Dict					       =					[]
      # Until we unindent, add backend objects to the list
      while len(lines[line_index]    ) <= 1 or lines[line_index].startswith(' ' * 8    ):
        a						: List[Any]					       =					lines[line_index]
        a						: 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(' ' * 12    ):
          objects.append(line[12:-2]    )
        line_index += 1

      a						: Tuple					       =					objects
    else:
      line_index += 1

  return import_dict_objects, type_hint_objects
def 				lowerCamelCase__	(						_A			,      _A    ):

  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!"]

  a						: Dict					       =					[]
  for key in import_dict_objects.keys():
    a						: Tuple					       =					find_duplicates(import_dict_objects[key]    )
    if duplicate_imports:
      errors.append(f"""Duplicate _import_structure definitions for: {duplicate_imports}"""    )
    a						: Dict					       =					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]    )    ):
      a						: Dict					       =					'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 				lowerCamelCase__	(						):
  a						: Dict					       =					[]
  for root, _, files in os.walk(_A    ):
    if "__init__.py" in files:
      a						: Tuple					       =					os.path.join(_A			,      '__init__.py'    )
      a						: Dict					       =					parse_init(_A    )
      if objects is not None:
        a						: Optional[Any]					       =					analyze_results(*_A    )
        if len(_A    ) > 0:
          a						: List[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 				lowerCamelCase__	(						):
  a						: str					       =					[]
  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
      a						: Dict					       =					str((Path(_A    ) / folder).relative_to(_A    )    )
      a						: Any					       =					short_path.replace(os.path.sep			,      '.'    )
      submodules.append(_A    )
    for fname in files:
      if fname == "__init__.py":
        continue
      a						: int					       =					str((Path(_A    ) / fname).relative_to(_A    )    )
      a						: Dict					       =					short_path.replace('.py'			,      ''    ).replace(os.path.sep			,      '.'    )
      if len(submodule.split('.'    )    ) == 1:
        submodules.append(_A    )
  return submodules


lowerCAmelCase: Dict							=						[
    'convert_pytorch_checkpoint_to_tf2',
    'modeling_flax_pytorch_utils',
]
def 				lowerCamelCase__	(						):
  # This is to make sure the transformers module imported is the one in the repo.
  a						: Dict					       =					importlib.util.spec_from_file_location(
      'transformers'			,      os.path.join(_A			,      '__init__.py'    )			,      submodule_search_locations=[PATH_TO_TRANSFORMERS]			,      )
  a						: Union[str, Any]					       =					spec.loader.load_module()

  a						: int					       =					[
      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:
    a						: Optional[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()




'''simple docstring'''






import argparse
import os
import re

import packaging.version


lowerCAmelCase: List[str]							=						'examples/'
lowerCAmelCase: List[Any]							=						{
    'examples': (re.compile(r'^check_min_version\("[^"]+"\)\s*$', re.MULTILINE), 'check_min_version("VERSION")\n'),
    'init': (re.compile(r'^__version__\s+=\s+"([^"]+)"\s*$', re.MULTILINE), '__version__ = "VERSION"\n'),
    'setup': (re.compile(r'^(\s*)version\s*=\s*"[^"]+",', re.MULTILINE), r'\1version="VERSION",'),
    'doc': (re.compile(r'^(\s*)release\s*=\s*"[^"]+"$', re.MULTILINE), 'release = "VERSION"\n'),
}
lowerCAmelCase: str							=						{
    'init': 'src/transformers/__init__.py',
    'setup': 'setup.py',
}
lowerCAmelCase: str							=						'README.md'
def 				lowerCamelCase__	(						_A			,      _A			,      _A    ):
  with open(_A			,      'r'			,      encoding='utf-8'			,      newline='\n'    ) as f:
    a						: Tuple					       =					f.read()
  a       ,     a						: Tuple					       =					REPLACE_PATTERNS[pattern]
  a						: Dict					       =					replace.replace('VERSION'			,      _A    )
  a						: Dict					       =					re_pattern.sub(_A			,      _A    )
  with open(_A			,      'w'			,      encoding='utf-8'			,      newline='\n'    ) as f:
    f.write(_A    )
def 				lowerCamelCase__	(						_A    ):
  for folder, directories, fnames in os.walk(_A    ):
    # Removing some of the folders with non-actively maintained examples from the walk
    if "research_projects" in directories:
      directories.remove('research_projects'    )
    if "legacy" in directories:
      directories.remove('legacy'    )
    for fname in fnames:
      if fname.endswith('.py'    ):
        update_version_in_file(os.path.join(_A			,      _A    )			,      _A			,      pattern='examples'    )
def 				lowerCamelCase__	(						_A			,      _A=False    ):
  for pattern, fname in REPLACE_FILES.items():
    update_version_in_file(_A			,      _A			,      _A    )
  if not patch:
    update_version_in_examples(_A    )
def 				lowerCamelCase__	(						):
  a						: Tuple					       =					'🤗 Transformers currently provides the following architectures'
  a						: Any					       =					'1. Want to contribute a new model?'
  with open(_A			,      'r'			,      encoding='utf-8'			,      newline='\n'    ) as f:
    a						: Tuple					       =					f.readlines()

  # Find the start of the list.
  a						: Optional[int]					       =					0
  while not lines[start_index].startswith(_start_prompt    ):
    start_index += 1
  start_index += 1

  a						: Optional[int]					       =					start_index
  # Update the lines in the model list.
  while not lines[index].startswith(_end_prompt    ):
    if lines[index].startswith('1.'    ):
      a						: List[Any]					       =					lines[index].replace(
          'https://huggingface.co/docs/transformers/main/model_doc'			,      'https://huggingface.co/docs/transformers/model_doc'			,      )
    index += 1

  with open(_A			,      'w'			,      encoding='utf-8'			,      newline='\n'    ) as f:
    f.writelines(_A    )
def 				lowerCamelCase__	(						):
  with open(REPLACE_FILES['init']			,      'r'    ) as f:
    a						: Union[str, Any]					       =					f.read()
  a						: Tuple					       =					REPLACE_PATTERNS['init'][0].search(_A    ).groups()[0]
  return packaging.version.parse(_A    )
def 				lowerCamelCase__	(						_A=False    ):
  a						: int					       =					get_version()
  if patch and default_version.is_devrelease:
    raise ValueError('Can\'t create a patch version from the dev branch, checkout a released version!'    )
  if default_version.is_devrelease:
    a						: Any					       =					default_version.base_version
  elif patch:
    a						: Dict					       =					f"""{default_version.major}.{default_version.minor}.{default_version.micro + 1}"""
  else:
    a						: Union[str, Any]					       =					f"""{default_version.major}.{default_version.minor + 1}.0"""

  # Now let's ask nicely if that's the right one.
  a						: List[Any]					       =					input(f"""Which version are you releasing? [{default_version}]"""    )
  if len(_A    ) == 0:
    a						: Union[str, Any]					       =					default_version

  print(f"""Updating version to {version}."""    )
  global_version_update(_A			,      patch=_A    )
  if not patch:
    print('Cleaning main README, don\'t forget to run `make fix-copies`.'    )
    clean_main_ref_in_model_list()
def 				lowerCamelCase__	(						):
  a						: int					       =					get_version()
  a						: Any					       =					f"""{current_version.major}.{current_version.minor + 1}.0.dev0"""
  a						: int					       =					current_version.base_version

  # Check with the user we got that right.
  a						: Tuple					       =					input(f"""Which version are we developing now? [{dev_version}]"""    )
  if len(_A    ) == 0:
    a						: Optional[int]					       =					dev_version

  print(f"""Updating version to {version}."""    )
  global_version_update(_A    )
  print('Cleaning main README, don\'t forget to run `make fix-copies`.'    )
  clean_main_ref_in_model_list()


if __name__ == "__main__":
     lowerCAmelCase: Tuple							=						argparse.ArgumentParser()
     parser.add_argument('--post_release', action='store_true', help='Whether this is pre or post release.')
     parser.add_argument('--patch', action='store_true', help='Whether or not this is a patch release.')
     lowerCAmelCase: Optional[Any]							=						parser.parse_args()
     if not args.post_release:
          pre_release_work(patch=args.patch)
     elif args.patch:
          print('Nothing to do after a patch :-)')
     else:
          post_release_work()