



import json
import os
import tempfile

import datasets
from utils import generate_example_dataset, get_duration


UpperCamelCase__			=    50_00_00

UpperCamelCase__			, UpperCamelCase__			=    os.path.split(__file__)
UpperCamelCase__			=    os.path.join(RESULTS_BASEPATH, "results", RESULTS_FILENAME.replace(".py", ".json"))






@get_duration
def 			UpperCAmelCase__     (     lowercase__ )			->   Any:
						__lowercase											=					dataset.select(range(0    ,							len(lowercase__ )    ,							2 ) )






@get_duration
def 			UpperCAmelCase__     (     lowercase__ )			->   str:
						__lowercase											=					dataset.sort("""numbers""" )






@get_duration
def 			UpperCAmelCase__     (     lowercase__ )			->   int:
						__lowercase											=					dataset.shuffle()






@get_duration
def 			UpperCAmelCase__     (     lowercase__ )			->   Tuple:
						__lowercase											=					dataset.train_test_split(0.1 )






@get_duration
def 			UpperCAmelCase__     (     lowercase__    ,							lowercase__=10 )			->   Any:
						for shard_id in range(lowercase__ ):
												__lowercase											=					dataset.shard(lowercase__    ,							lowercase__ )






def 			UpperCAmelCase__     (     )			->   Dict:
						__lowercase											=					{"""num examples""": SPEED_TEST_N_EXAMPLES}
						__lowercase											=					(select, sort, shuffle, train_test_split, shard)
						with tempfile.TemporaryDirectory() as tmp_dir:
												print("""generating dataset""" )
												__lowercase											=					datasets.Features({"""text""": datasets.Value("""string""" ), """numbers""": datasets.Value("""float32""" )} )
												__lowercase											=					generate_example_dataset(
												    os.path.join(lowercase__    ,							"""dataset.arrow""" )    ,							lowercase__    ,							num_examples=lowercase__ )
												print("""Functions""" )
												for func in functions:
																		print(func.__name__ )
																		__lowercase											=					func(lowercase__ )

						with open(lowercase__    ,							"""wb""" ) as f:
												f.write(json.dumps(lowercase__ ).encode("""utf-8""" ) )


if __name__ == "__main__":  # useful to run the profiler
			benchmark_indices_mapping()










import argparse
import os
import re


# All paths are set with the intent you should run this script from the root of the repo with the command
# python utils/check_dummies.py
UpperCamelCase__			=    "src/transformers"

# Matches is_xxx_available()
UpperCamelCase__			=    re.compile(R"is\_([a-z_]*)_available()")
# Matches from xxx import bla
UpperCamelCase__			=    re.compile(R"\s+from\s+\S*\s+import\s+([^\(\s].*)\n")
UpperCamelCase__			=    re.compile(R"^\s+if\s+not\s+\(?is\_[a-z_]*\_available\(\)")


UpperCamelCase__			=    "\n{0} = None\n"

UpperCamelCase__			=    "\nclass {0}(metaclass=DummyObject):\n    _backends = {1}\n\n    def __init__(self, *args, **kwargs):\n        requires_backends(self, {1})\n"


UpperCamelCase__			=    "\ndef {0}(*args, **kwargs):\n    requires_backends({0}, {1})\n"






def 			UpperCAmelCase__     (     lowercase__ )			->   List[Any]:
						if _re_test_backend.search(lowercase__ ) is None:
												return None
						__lowercase											=					[b[0] for b in _re_backend.findall(lowercase__ )]
						backends.sort()
						return "_and_".join(lowercase__ )






def 			UpperCAmelCase__     (     )			->   Dict:
						with open(os.path.join(lowercase__    ,							"""__init__.py""" )    ,							"""r"""    ,							encoding="""utf-8"""    ,							newline="""\n""" ) as f:
												__lowercase											=					f.readlines()

						# Get to the point we do the actual imports for type checking
						__lowercase											=					0
						while not lines[line_index].startswith("""if TYPE_CHECKING""" ):
												line_index += 1

						__lowercase											=					{}
						# Go through the end of the file
						while line_index < len(lowercase__ ):
												# If the line is an if is_backend_available, we grab all objects associated.
												__lowercase											=					find_backend(lines[line_index] )
												if backend is not None:
																		while not lines[line_index].startswith("""    else:""" ):
																								line_index += 1
																		line_index += 1

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

																		__lowercase											=					objects
												else:
																		line_index += 1

						return backend_specific_objects






def 			UpperCAmelCase__     (     lowercase__    ,							lowercase__ )			->   int:
						if name.isupper():
												return DUMMY_CONSTANT.format(lowercase__ )
						elif name.islower():
												return DUMMY_FUNCTION.format(lowercase__    ,							lowercase__ )
						else:
												return DUMMY_CLASS.format(lowercase__    ,							lowercase__ )






def 			UpperCAmelCase__     (     lowercase__=None )			->   List[str]:
						if backend_specific_objects is None:
												__lowercase											=					read_init()
						# For special correspondence backend to module name as used in the function requires_modulename
						__lowercase											=					{}

						for backend, objects in backend_specific_objects.items():
												__lowercase											=					"""[""" + """, """.join(F"\"{b}\"" for b in backend.split("""_and_""" ) ) + """]"""
												__lowercase											=					"""# This file is autogenerated by the command `make fix-copies`, do not edit.\n"""
												dummy_file += "from ..utils import DummyObject, requires_backends\n\n"
												dummy_file += "\n".join([create_dummy_object(lowercase__    ,							lowercase__ ) for o in objects] )
												__lowercase											=					dummy_file

						return dummy_files






def 			UpperCAmelCase__     (     lowercase__=False )			->   Optional[Any]:
						__lowercase											=					create_dummy_files()
						# For special correspondence backend to shortcut as used in utils/dummy_xxx_objects.py
						__lowercase											=					{"""torch""": """pt"""}

						# Locate actual dummy modules and read their content.
						__lowercase											=					os.path.join(lowercase__    ,							"""utils""" )
						__lowercase											=					{
						    backend: os.path.join(lowercase__    ,							F"dummy_{short_names.get(lowercase__    ,							lowercase__ )}_objects.py" )
						    for backend in dummy_files.keys()
						}

						__lowercase											=					{}
						for backend, file_path in dummy_file_paths.items():
												if os.path.isfile(lowercase__ ):
																		with open(lowercase__    ,							"""r"""    ,							encoding="""utf-8"""    ,							newline="""\n""" ) as f:
																								__lowercase											=					f.read()
												else:
																		__lowercase											=					""""""

						for backend in dummy_files.keys():
												if dummy_files[backend] != actual_dummies[backend]:
																		if overwrite:
																								print(
																								    F"Updating transformers.utils.dummy_{short_names.get(lowercase__    ,							lowercase__ )}_objects.py as the main "
																								    """__init__ has new objects.""" )
																								with open(dummy_file_paths[backend]    ,							"""w"""    ,							encoding="""utf-8"""    ,							newline="""\n""" ) as f:
																														f.write(dummy_files[backend] )
																		else:
																								raise ValueError(
																								    """The main __init__ has objects that are not present in """
																								    F"transformers.utils.dummy_{short_names.get(lowercase__    ,							lowercase__ )}_objects.py. Run `make fix-copies` "
																								    """to fix this.""" )


if __name__ == "__main__":
			UpperCamelCase__			=    argparse.ArgumentParser()
			parser.add_argument("--fix_and_overwrite", action="store_true", help="Whether to fix inconsistencies.")
			UpperCamelCase__			=    parser.parse_args()

			check_dummies(args.fix_and_overwrite)





