






'''simple docstring'''





import argparse
from collections import defaultdict

import yaml


_UpperCamelCase						:						Union[str, Any]  =					'docs/source/en/_toctree.yml'
def __UpperCAmelCase   (					A					:			Tuple		)    ->   Optional[Any]:
			UpperCAmelCase_      :       Union[str, Any]          =       defaultdict(A		)
			UpperCAmelCase_      :       Optional[Any]          =       []
			UpperCAmelCase_      :       Optional[int]          =       []
			for doc in doc_list:
						if "local" in doc:
									counts[doc["local"]] += 1

						if doc["title"].lower() == "overview":
									overview_doc.append({'''local''': doc['''local'''], '''title''': doc['''title''']}		)
						else:
									new_doc_list.append(A		)

			UpperCAmelCase_      :       int          =       new_doc_list
			UpperCAmelCase_      :       List[Any]          =       [key for key, value in counts.items() if value > 1]

			UpperCAmelCase_      :       Optional[int]          =       []
			for duplicate_key in duplicates:
						UpperCAmelCase_      :       Optional[int]          =       list({doc['''title'''] for doc in doc_list if doc['''local'''] == duplicate_key}		)
						if len(A		) > 1:
									raise ValueError(
									    F"{duplicate_key} is present several times in the documentation table of content at "
									    '''`docs/source/en/_toctree.yml` with different *Title* values. Choose one of those and remove the '''
									    '''others.'''		)
						# Only add this once
						new_doc.append({'''local''': duplicate_key, '''title''': titles[0]}		)

			# Add none duplicate-keys
			new_doc.extend([doc for doc in doc_list if '''local''' not in counts or counts[doc['''local''']] == 1]		)
			UpperCAmelCase_      :       Optional[Any]          =       sorted(A			,       key=lambda A		: s["title"].lower()		)

			# "overview" gets special treatment and is always first
			if len(A		) > 1:
						raise ValueError('''{doc_list} has two \'overview\' docs which is not allowed.'''		)

			overview_doc.extend(A		)

			# Sort
			return overview_doc
def __UpperCAmelCase   (					A					:			List[str]=False		)    ->   Optional[Any]:
			with open(A			,       encoding='''utf-8'''		) as f:
						UpperCAmelCase_      :       Dict          =       yaml.safe_load(f.read()		)

			# Get to the API doc
			UpperCAmelCase_      :       Union[str, Any]          =       0
			while content[api_idx]["title"] != "API":
						api_idx += 1
			UpperCAmelCase_      :       str          =       content[api_idx]['''sections''']

			# Then to the model doc
			UpperCAmelCase_      :       Optional[Any]          =       0
			while api_doc[scheduler_idx]["title"] != "Schedulers":
						scheduler_idx += 1

			UpperCAmelCase_      :       int          =       api_doc[scheduler_idx]['''sections''']
			UpperCAmelCase_      :       Dict          =       clean_doc_toc(A		)

			UpperCAmelCase_      :       List[Any]          =       False
			if new_scheduler_doc != scheduler_doc:
						UpperCAmelCase_      :       Optional[int]          =       True
						if overwrite:
									UpperCAmelCase_      :       Union[str, Any]          =       new_scheduler_doc

			if diff:
						if overwrite:
									UpperCAmelCase_      :       Any          =       api_doc
									with open(A			,       '''w'''			,       encoding='''utf-8'''		) as f:
												f.write(yaml.dump(A			,       allow_unicode=A		)		)
						else:
									raise ValueError(
									    '''The model doc part of the table of content is not properly sorted, run `make style` to fix this.'''		)
def __UpperCAmelCase   (					A					:			Tuple=False		)    ->   Dict:
			with open(A			,       encoding='''utf-8'''		) as f:
						UpperCAmelCase_      :       Dict          =       yaml.safe_load(f.read()		)

			# Get to the API doc
			UpperCAmelCase_      :       Dict          =       0
			while content[api_idx]["title"] != "API":
						api_idx += 1
			UpperCAmelCase_      :       List[Any]          =       content[api_idx]['''sections''']

			# Then to the model doc
			UpperCAmelCase_      :       Optional[Any]          =       0
			while api_doc[pipeline_idx]["title"] != "Pipelines":
						pipeline_idx += 1

			UpperCAmelCase_      :       Dict          =       False
			UpperCAmelCase_      :       Dict          =       api_doc[pipeline_idx]['''sections''']
			UpperCAmelCase_      :       str          =       []

			# sort sub pipeline docs
			for pipeline_doc in pipeline_docs:
						if "section" in pipeline_doc:
									UpperCAmelCase_      :       List[Any]          =       pipeline_doc['''section''']
									UpperCAmelCase_      :       List[str]          =       clean_doc_toc(A		)
									if overwrite:
												UpperCAmelCase_      :       Tuple          =       new_sub_pipeline_doc
						new_pipeline_docs.append(A		)

			# sort overall pipeline doc
			UpperCAmelCase_      :       List[Any]          =       clean_doc_toc(A		)

			if new_pipeline_docs != pipeline_docs:
						UpperCAmelCase_      :       List[Any]          =       True
						if overwrite:
									UpperCAmelCase_      :       List[str]          =       new_pipeline_docs

			if diff:
						if overwrite:
									UpperCAmelCase_      :       Optional[int]          =       api_doc
									with open(A			,       '''w'''			,       encoding='''utf-8'''		) as f:
												f.write(yaml.dump(A			,       allow_unicode=A		)		)
						else:
									raise ValueError(
									    '''The model doc part of the table of content is not properly sorted, run `make style` to fix this.'''		)


if __name__ == "__main__":
						_UpperCamelCase						:						Union[str, Any]  =					argparse.ArgumentParser()
						parser.add_argument('--fix_and_overwrite', action='store_true', help='Whether to fix inconsistencies.')
						_UpperCamelCase						:						str  =					parser.parse_args()

						check_scheduler_doc(args.fix_and_overwrite)
						check_pipeline_doc(args.fix_and_overwrite)










'''simple docstring'''





import collections
import os
import re
from pathlib import Path


_UpperCamelCase						:						Optional[int]  =					'src/transformers'


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

			UpperCAmelCase_      :       List[str]          =       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
			UpperCAmelCase_      :       Tuple          =       []
			while not lines[line_index].startswith('''if TYPE_CHECKING'''		) and find_backend(lines[line_index]		) is None:
						UpperCAmelCase_      :       List[Any]          =       lines[line_index]
						# If we have everything on a single line, let's deal with it.
						if _re_one_line_import_struct.search(A		):
									UpperCAmelCase_      :       str          =       _re_one_line_import_struct.search(A		).groups()[0]
									UpperCAmelCase_      :       Any          =       re.findall(r'''\[([^\]]+)\]'''			,       A		)
									for imp in imports:
												objects.extend([obj[1:-1] for obj in imp.split(''', '''		)]		)
									line_index += 1
									continue
						UpperCAmelCase_      :       Dict          =       _re_import_struct_key_value.search(A		)
						if single_line_import_search is not None:
									UpperCAmelCase_      :       List[str]          =       [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

			UpperCAmelCase_      :       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.
						UpperCAmelCase_      :       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:
									UpperCAmelCase_      :       int          =       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

									UpperCAmelCase_      :       str          =       []
									# Until we unindent, add backend objects to the list
									while len(lines[line_index]		) <= 1 or lines[line_index].startswith(''' ''' * 4		):
												UpperCAmelCase_      :       Optional[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:
															UpperCAmelCase_      :       Dict          =       _re_import_struct_add_many.search(A		).groups()[0].split(''', '''		)
															UpperCAmelCase_      :       Dict          =       [obj[1:-1] for obj in imports if len(A		) > 0]
															objects.extend(A		)
												elif _re_between_brackets.search(A		) is not None:
															UpperCAmelCase_      :       int          =       _re_between_brackets.search(A		).groups()[0].split(''', '''		)
															UpperCAmelCase_      :       List[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

									UpperCAmelCase_      :       int          =       objects
						else:
									line_index += 1

    # At this stage we are in the TYPE_CHECKING part, first grab the objects without a specific backend
			UpperCAmelCase_      :       Union[str, Any]          =       []
			while (
			    line_index < len(A		)
			    and find_backend(lines[line_index]		) is None
			    and not lines[line_index].startswith('''else'''		)
			):
						UpperCAmelCase_      :       int          =       lines[line_index]
						UpperCAmelCase_      :       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(''' ''' * 8		):
									objects.append(line[8:-2]		)
						line_index += 1

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

									UpperCAmelCase_      :       str          =       []
									# Until we unindent, add backend objects to the list
									while len(lines[line_index]		) <= 1 or lines[line_index].startswith(''' ''' * 8		):
												UpperCAmelCase_      :       Tuple          =       lines[line_index]
												UpperCAmelCase_      :       Tuple          =       _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

									UpperCAmelCase_      :       Optional[Any]          =       objects
						else:
									line_index += 1

			return import_dict_objects, type_hint_objects
def __UpperCAmelCase   (					A					:			Dict			,       A					:			Union[str, Any]		)    ->   Optional[Any]:

			def find_duplicates(A					:			Optional[int]		):
						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!"]

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


_UpperCamelCase						:						List[Any]  =					[
    'convert_pytorch_checkpoint_to_tf2',
    'modeling_flax_pytorch_utils',
    'models.esm.openfold_utils',
]
def __UpperCAmelCase   (					)    ->   int:
			# This is to make sure the transformers module imported is the one in the repo.
			from transformers.utils import direct_transformers_import

			UpperCAmelCase_      :       Dict          =       direct_transformers_import(A		)

			UpperCAmelCase_      :       List[str]          =       set(transformers._import_structure.keys()		)
			# This contains all the base keys of the _import_structure object defined in the init, but if the user is missing
			# some optional dependencies, they may not have all of them. Thus we read the init to read all additions and
			# (potentiall re-) add them.
			with open(os.path.join(A			,       '''__init__.py'''		)			,       '''r'''		) as f:
						UpperCAmelCase_      :       Dict          =       f.read()
			import_structure_keys.update(set(re.findall(r'''import_structure\[\"([^\"]*)\"\]'''			,       A		)		)		)

			UpperCAmelCase_      :       Union[str, Any]          =       [
			    module
			    for module in get_transformers_submodules()
			    if module not in IGNORE_SUBMODULES and module not in import_structure_keys
			]

			if len(A		) > 0:
						UpperCAmelCase_      :       Optional[Any]          =       '''\n'''.join(F"- {module}" for module in module_not_registered		)
						raise ValueError(
						    '''The following submodules are not properly registed 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()


