code: 
  '''simple docstring'''
  import json
  import os
  import tempfile
  
  import datasets
  from utils import generate_example_dataset, get_duration
  
  
  lowercase__						=			5_0_0_0_0_0
  
  lowercase__					,     lowercase__						=			os.path.split(__file__)
  lowercase__						=			os.path.join(RESULTS_BASEPATH, "results", RESULTS_FILENAME.replace(".py", ".json"))
  
  
  
  
  @get_duration
  def       _UpperCamelCase	(					SCREAMING_SNAKE_CASE__ )     ->       int:
  
  
  		'''simple docstring'''
  
  
  		snake_case     :					Dict     =	dataset.select(range(0   ,					len(SCREAMING_SNAKE_CASE__ )   ,					2 ) )
  
  
  
  
  @get_duration
  def       _UpperCamelCase	(					SCREAMING_SNAKE_CASE__ )     ->       Dict:
  
  
  		'''simple docstring'''
  
  
  		snake_case     :					Any     =	dataset.sort('''numbers''' )
  
  
  
  
  @get_duration
  def       _UpperCamelCase	(					SCREAMING_SNAKE_CASE__ )     ->       List[str]:
  
  
  		'''simple docstring'''
  
  
  		snake_case     :					List[Any]     =	dataset.shuffle()
  
  
  
  
  @get_duration
  def       _UpperCamelCase	(					SCREAMING_SNAKE_CASE__ )     ->       Union[str, Any]:
  
  
  		'''simple docstring'''
  
  
  		snake_case     :					str     =	dataset.train_test_split(0.1 )
  
  
  
  
  @get_duration
  def       _UpperCamelCase	(					SCREAMING_SNAKE_CASE__   ,					SCREAMING_SNAKE_CASE__=10 )     ->       List[Any]:
  
  
  		'''simple docstring'''
  
  
  		for shard_id in range(SCREAMING_SNAKE_CASE__ ):
  				snake_case     :					List[Any]     =	dataset.shard(SCREAMING_SNAKE_CASE__   ,					SCREAMING_SNAKE_CASE__ )
  
  
  
  
  def       _UpperCamelCase	(					)     ->       Dict:
  
  
  		'''simple docstring'''
  
  
  		snake_case     :					int     =	{'''num examples''': SPEED_TEST_N_EXAMPLES}
  		snake_case     :					Any     =	(select, sort, shuffle, train_test_split, shard)
  		with tempfile.TemporaryDirectory() as tmp_dir:
  				print('''generating dataset''' )
  				snake_case     :					Union[str, Any]     =	datasets.Features({'''text''': datasets.Value('''string''' ), '''numbers''': datasets.Value('''float32''' )} )
  				snake_case     :					List[Any]     =	generate_example_dataset(
  				    os.path.join(SCREAMING_SNAKE_CASE__   ,					'''dataset.arrow''' )   ,					SCREAMING_SNAKE_CASE__   ,					num_examples=SCREAMING_SNAKE_CASE__ )
  				print('''Functions''' )
  				for func in functions:
  						print(func.__name__ )
  						snake_case     :					int     =	func(SCREAMING_SNAKE_CASE__ )
  
  		with open(SCREAMING_SNAKE_CASE__   ,					'''wb''' ) as f:
  				f.write(json.dumps(SCREAMING_SNAKE_CASE__ ).encode('''utf-8''' ) )
  
  
  if __name__ == "__main__":  # useful to run the profiler
  					benchmark_indices_mapping()
  
  
  
style_context: 
  '''simple docstring'''
  def       _UpperCamelCase	(					SCREAMING_SNAKE_CASE__   ,					SCREAMING_SNAKE_CASE__ )     ->       int:
  
  
  		'''simple docstring'''
  
  
  		snake_case     :					Optional[Any]     =	0
  		for ele in arr:
  				total_sum += product_sum(SCREAMING_SNAKE_CASE__   ,					depth + 1 ) if isinstance(SCREAMING_SNAKE_CASE__   ,					SCREAMING_SNAKE_CASE__ ) else ele
  		return total_sum * depth
  
  
  
  
  def       _UpperCamelCase	(					SCREAMING_SNAKE_CASE__ )     ->       int:
  
  
  		'''simple docstring'''
  
  
  		return product_sum(SCREAMING_SNAKE_CASE__   ,					1 )
  
  
  if __name__ == "__main__":
  						import doctest
  
  						doctest.testmod()
  
  
  