





import argparse
import os

from accelerate.utils import ComputeEnvironment

from .cluster import get_cluster_input
from .config_args import cache_dir, default_config_file, default_yaml_config_file, load_config_from_file  # noqa: F401
from .config_utils import _ask_field, _ask_options, _convert_compute_environment  # noqa: F401
from .sagemaker import get_sagemaker_input


lowercase    : Optional[int]    	=  """Launches a series of prompts to create and save a `default_config.yaml` configuration file for your training system. Should always be ran first on your machine"""


def     SCREAMING_SNAKE_CASE			(      ):

       '''simple docstring'''





       _UpperCamelCase              =						_ask_options(
           '''In which compute environment are you running?'''    ,       ['''This machine''', '''AWS (Amazon SageMaker)''']    ,       _convert_compute_environment    ,       )
       if compute_environment == ComputeEnvironment.AMAZON_SAGEMAKER:
              _UpperCamelCase              =						get_sagemaker_input()
       else:
              _UpperCamelCase              =						get_cluster_input()
       return config


def     SCREAMING_SNAKE_CASE			(      lowerCAmelCase=None					):

       '''simple docstring'''





       if subparsers is not None:
              _UpperCamelCase              =						subparsers.add_parser('''config'''    ,       description=lowerCAmelCase					)
       else:
              _UpperCamelCase              =						argparse.ArgumentParser('''Accelerate config command'''    ,       description=lowerCAmelCase					)

       parser.add_argument(
           '''--config_file'''    ,       default=lowerCAmelCase    ,       help=(
               '''The path to use to store the config file. Will default to a file named default_config.yaml in the cache '''
               '''location, which is the content of the environment `HF_HOME` suffixed with \'accelerate\', or if you don\'t have '''
               '''such an environment variable, your cache directory (\'~/.cache\' or the content of `XDG_CACHE_HOME`) suffixed '''
               '''with \'huggingface\'.'''
           )    ,       )

       if subparsers is not None:
              parser.set_defaults(func=lowerCAmelCase					)
       return parser


def     SCREAMING_SNAKE_CASE			(      lowerCAmelCase					):

       '''simple docstring'''





       _UpperCamelCase              =						get_user_input()
       if args.config_file is not None:
              _UpperCamelCase              =						args.config_file
       else:
              if not os.path.isdir(lowerCAmelCase					):
                     os.makedirs(lowerCAmelCase					)
              _UpperCamelCase              =						default_yaml_config_file

       if config_file.endswith('''.json'''					):
              config.to_json_file(lowerCAmelCase					)
       else:
              config.to_yaml_file(lowerCAmelCase					)
       print(f'''accelerate configuration saved at {config_file}'''					)


def     SCREAMING_SNAKE_CASE			(      ):

       '''simple docstring'''





       _UpperCamelCase              =						config_command_parser()
       _UpperCamelCase              =						parser.parse_args()
       config_command(lowerCAmelCase					)


if __name__ == "__main__":
       main()













def     SCREAMING_SNAKE_CASE			(      lowerCAmelCase    ,       lowerCAmelCase    ,       lowerCAmelCase = 0    ,       lowerCAmelCase = 0					):
       _UpperCamelCase              =						right or len(lowerCAmelCase					) - 1
       if left > right:
              return -1
       elif list_data[left] == key:
              return left
       elif list_data[right] == key:
              return right
       else:
              return search(lowerCAmelCase    ,       lowerCAmelCase    ,       left + 1    ,       right - 1					)


if __name__ == "__main__":
       import doctest

       doctest.testmod()






