



'''simple docstring'''



import collections
import os
import re
from pathlib import Path


lowerCAmelCase	:		int				=   '''src/transformers'''


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



def   __lowerCAmelCase					(							lowerCamelCase   :						Union[str, Any]					):






	'''simple docstring'''




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



def   __lowerCAmelCase					(							lowerCamelCase   :						Any					):






	'''simple docstring'''




	with open(lowerCamelCase				,		"r"				,		encoding="utf-8"				,		newline="\n"					) as f:
		__lowerCAmelCase												=  f.readlines()

	__lowerCAmelCase												=  0
	while line_index < len(lowerCamelCase					) and not lines[line_index].startswith("_import_structure = {"					):
		line_index += 1

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

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

	__lowerCAmelCase												=  {"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.
		__lowerCAmelCase												=  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:
			__lowerCAmelCase												=  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

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

			__lowerCAmelCase												=  objects
		else:
			line_index += 1

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

	__lowerCAmelCase												=  {"none": objects}
	# Let's continue with backend-specific objects
	while line_index < len(lowerCamelCase					):
		# If the line is an if is_backend_available, we grab all objects associated.
		__lowerCAmelCase												=  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:
			__lowerCAmelCase												=  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

			__lowerCAmelCase												=  []
			# Until we unindent, add backend objects to the list
			while len(lines[line_index]					) <= 1 or lines[line_index].startswith(" " * 8					):
				__lowerCAmelCase												=  lines[line_index]
				__lowerCAmelCase												=  _re_import.search(lowerCamelCase					)
				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

			__lowerCAmelCase												=  objects
		else:
			line_index += 1

	return import_dict_objects, type_hint_objects



def   __lowerCAmelCase					(							lowerCamelCase   :						str				,		lowerCamelCase   :						Dict					):






	'''simple docstring'''




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

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






	'''simple docstring'''




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



def   __lowerCAmelCase					(							):






	'''simple docstring'''




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


lowerCAmelCase	:		List[str]				=   [
    '''convert_pytorch_checkpoint_to_tf2''',
    '''modeling_flax_pytorch_utils''',
    '''models.esm.openfold_utils''',
]



def   __lowerCAmelCase					(							):






	'''simple docstring'''




	from transformers.utils import direct_transformers_import

	__lowerCAmelCase												=  direct_transformers_import(lowerCamelCase					)

	__lowerCAmelCase												=  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(lowerCamelCase				,		"__init__.py"					)				,		"r"					) as f:
		__lowerCAmelCase												=  f.read()
	import_structure_keys.update(set(re.findall(r"import_structure\[\"([^\"]*)\"\]"				,		lowerCamelCase					)					)					)

	__lowerCAmelCase												=  [
	    module
	    for module in get_transformers_submodules()
	    if module not in IGNORE_SUBMODULES and module not in import_structure_keys
	]

	if len(lowerCamelCase					) > 0:
		__lowerCAmelCase												=  "\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()




'''simple docstring'''



import importlib
import os
from dataclasses import dataclass
from enum import Enum
from typing import Any, Dict, Optional, Union

import torch

from ..utils import BaseOutput


lowerCAmelCase	:		Optional[Any]				=   '''scheduler_config.json'''






class 					UpperCAmelCase__	(		UpperCamelCase__ ):
					a      :     str        =							1
					a      :     Optional[int]        =							2
					a      :     int        =							3
					a      :     Union[str, Any]        =							4
					a      :     int        =							5
					a      :     Optional[int]        =							6
					a      :     str        =							7
					a      :     List[Any]        =							8
					a      :     List[str]        =							9
					a      :     List[str]        =							1_0
					a      :     int        =							1_1
					a      :     Any        =							1_2
					a      :     Any        =							1_3
					a      :     Tuple        =							1_4






@dataclass
class 					UpperCAmelCase__	(		UpperCamelCase__ ):
					a      :     torch.FloatTensor






class 					UpperCAmelCase__	:
					a      :     Tuple        =							SCHEDULER_CONFIG_NAME
					a      :     Union[str, Any]        =							[]
					a      :     str        =							True



					@classmethod
					def        UpperCAmelCase_ (    cls			,				UpperCamelCase = None			,				UpperCamelCase = None			,				UpperCamelCase=False			,				**UpperCamelCase			,				)     ->			int:
						__lowerCAmelCase							,       __lowerCAmelCase							,       __lowerCAmelCase												=  cls.load_config(
						    pretrained_model_name_or_path=UpperCamelCase			,				subfolder=UpperCamelCase			,				return_unused_kwargs=UpperCamelCase			,				return_commit_hash=UpperCamelCase			,				**UpperCamelCase			,				)
						return cls.from_config(UpperCamelCase			,				return_unused_kwargs=UpperCamelCase			,				**UpperCamelCase     )



					def        UpperCAmelCase_ (    self			,				UpperCamelCase			,				UpperCamelCase = False			,				**UpperCamelCase     )     ->			Dict:
						self.save_config(save_directory=UpperCamelCase			,				push_to_hub=UpperCamelCase			,				**UpperCamelCase     )



					@property
					def        UpperCAmelCase_ (    self     )     ->			str:
						return self._get_compatibles()



					@classmethod
					def        UpperCAmelCase_ (    cls     )     ->			Tuple:
						__lowerCAmelCase												=  list(set([cls.__name__] + cls._compatibles     )     )
						__lowerCAmelCase												=  importlib.import_module(__name__.split("."     )[0]     )
						__lowerCAmelCase												=  [
						    getattr(UpperCamelCase			,				UpperCamelCase     ) for c in compatible_classes_str if hasattr(UpperCamelCase			,				UpperCamelCase     )
						]
						return compatible_classes