



# Copyright 2021 The HuggingFace Team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

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


lowerCAmelCase__					  =  '''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  __lowerCamelCase (       ):

				"""simple docstring"""


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


def  __lowerCamelCase (       lowerCamelCase__=None						):

				"""simple docstring"""


				if subparsers is not None:
								lowercase__					:   Tuple													=					subparsers.add_parser("config"					,			description=lowerCamelCase__						)
				else:
								lowercase__					:   List[str]													=					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  __lowerCamelCase (       lowerCamelCase__						):

				"""simple docstring"""


				lowercase__					:   Optional[Any]													=					get_user_input()
				if args.config_file is not None:
								lowercase__					:   Any													=					args.config_file
				else:
								if not os.path.isdir(lowerCamelCase__						):
												os.makedirs(lowerCamelCase__						)
								lowercase__					:   Optional[Any]													=					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  __lowerCamelCase (       ):

				"""simple docstring"""


				lowercase__					:   Union[str, Any]													=					config_command_parser()
				lowercase__					:   str													=					parser.parse_args()
				config_command(lowerCamelCase__						)


if __name__ == "__main__":
				main()









from collections.abc import Iterator, MutableMapping
from dataclasses import dataclass
from typing import Generic, TypeVar

lowerCAmelCase__					  =  TypeVar('''KEY''')
lowerCAmelCase__					  =  TypeVar('''VAL''')

@dataclass(frozen=_UpperCamelCase    ,      slots=_UpperCamelCase      )
class 				snake_case__(Generic[KEY, VAL]      ):

					"""simple docstring"""






					lowercase_								=		42
					lowercase_								=		42






class 				snake_case__(_Item      ):

					"""simple docstring"""






					def __init__(							self :     List[str]					):
									super().__init__(SCREAMING_SNAKE_CASE				,       SCREAMING_SNAKE_CASE					)







					def __bool__(							self :     Tuple					):
									return False






lowerCAmelCase__					  =  _DeletedItem()

class 				snake_case__(MutableMapping[KEY, VAL]      ):

					"""simple docstring"""






					def __init__(							self :     Any				,       SCREAMING_SNAKE_CASE :     int = 8				,       SCREAMING_SNAKE_CASE :     float = 0.75					):
									lowercase__					:   Any													=					initial_block_size
									lowercase__					:   list[_Item | None]													=					[None] * initial_block_size
									assert 0.0 < capacity_factor < 1.0
									lowercase__					:   Dict													=					capacity_factor
									lowercase__					:   Optional[int]													=					0







					def 			snake_case					(							self :     Optional[Any]				,       SCREAMING_SNAKE_CASE :     KEY					):
									return hash(SCREAMING_SNAKE_CASE					) % len(self._buckets					)







					def 			snake_case					(							self :     Optional[int]				,       SCREAMING_SNAKE_CASE :     int					):
									return (ind + 1) % len(self._buckets					)







					def 			snake_case					(							self :     Tuple				,       SCREAMING_SNAKE_CASE :     int				,       SCREAMING_SNAKE_CASE :     KEY				,       SCREAMING_SNAKE_CASE :     VAL					):
									lowercase__					:   Tuple													=					self._buckets[ind]
									if not stored:
													lowercase__					:   int													=					_Item(SCREAMING_SNAKE_CASE				,       SCREAMING_SNAKE_CASE					)
													self._len += 1
													return True
									elif stored.key == key:
													lowercase__					:   str													=					_Item(SCREAMING_SNAKE_CASE				,       SCREAMING_SNAKE_CASE					)
													return True
									else:
													return False







					def 			snake_case					(							self :     str					):
									lowercase__					:   str													=					len(self._buckets					) * self._capacity_factor
									return len(self					) >= int(SCREAMING_SNAKE_CASE					)







					def 			snake_case					(							self :     List[str]					):
									if len(self._buckets					) <= self._initial_block_size:
													return False
									lowercase__					:   Optional[Any]													=					len(self._buckets					) * self._capacity_factor / 2
									return len(self					) < limit







					def 			snake_case					(							self :     List[Any]				,       SCREAMING_SNAKE_CASE :     int					):
									lowercase__					:   Tuple													=					self._buckets
									lowercase__					:   Optional[int]													=					[None] * new_size
									lowercase__					:   int													=					0
									for item in old_buckets:
													if item:
																	self._add_item(item.key				,       item.val					)







					def 			snake_case					(							self :     int					):
									self._resize(len(self._buckets					) * 2					)







					def 			snake_case					(							self :     Optional[Any]					):
									self._resize(len(self._buckets					) // 2					)







					def 			snake_case					(							self :     List[Any]				,       SCREAMING_SNAKE_CASE :     KEY					):
									lowercase__					:   Tuple													=					self._get_bucket_index(SCREAMING_SNAKE_CASE					)
									for _ in range(len(self._buckets					)					):
													yield ind
													lowercase__					:   Union[str, Any]													=					self._get_next_ind(SCREAMING_SNAKE_CASE					)







					def 			snake_case					(							self :     str				,       SCREAMING_SNAKE_CASE :     KEY				,       SCREAMING_SNAKE_CASE :     VAL					):
									for ind in self._iterate_buckets(SCREAMING_SNAKE_CASE					):
													if self._try_set(SCREAMING_SNAKE_CASE				,       SCREAMING_SNAKE_CASE				,       SCREAMING_SNAKE_CASE					):
																	break







					def __setitem__(							self :     List[str]				,       SCREAMING_SNAKE_CASE :     KEY				,       SCREAMING_SNAKE_CASE :     VAL					):
									if self._is_full():
													self._size_up()

									self._add_item(SCREAMING_SNAKE_CASE				,       SCREAMING_SNAKE_CASE					)







					def __delitem__(							self :     int				,       SCREAMING_SNAKE_CASE :     KEY					):
									for ind in self._iterate_buckets(SCREAMING_SNAKE_CASE					):
													lowercase__					:   Union[str, Any]													=					self._buckets[ind]
													if item is None:
																	raise KeyError(SCREAMING_SNAKE_CASE					)
													if item is _deleted:
																	continue
													if item.key == key:
																	lowercase__					:   Optional[int]													=					_deleted
																	self._len -= 1
																	break
									if self._is_sparse():
													self._size_down()







					def __getitem__(							self :     Tuple				,       SCREAMING_SNAKE_CASE :     KEY					):
									for ind in self._iterate_buckets(SCREAMING_SNAKE_CASE					):
													lowercase__					:   Any													=					self._buckets[ind]
													if item is None:
																	break
													if item is _deleted:
																	continue
													if item.key == key:
																	return item.val
									raise KeyError(SCREAMING_SNAKE_CASE					)







					def __len__(							self :     Optional[Any]					):
									return self._len







					def __iter__(							self :     List[str]					):
									yield from (item.key for item in self._buckets if item)







					def __repr__(							self :     str					):
									lowercase__					:   int													=					" ,".join(
									    f"""{item.key}: {item.val}""" for item in self._buckets if item					)
									return f"""HashMap({val_string})"""




