"""simple docstring"""




# 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


_a				:		Tuple	   =       '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					(      )	->				Tuple:
  _lowerCAmelCase						:	Tuple        =					_ask_options(
      """In which compute environment are you running?"""      ,["""This machine""", """AWS (Amazon SageMaker)"""]      ,_convert_compute_environment      ,)
  if compute_environment == ComputeEnvironment.AMAZON_SAGEMAKER:
    _lowerCAmelCase						:	List[Any]        =					get_sagemaker_input()
  else:
    _lowerCAmelCase						:	List[Any]        =					get_cluster_input()
  return config





def 							SCREAMING_SNAKE_CASE					(      _lowerCamelCase						:	Optional[Any]=None     )	->				Dict:
  if subparsers is not None:
    _lowerCAmelCase						:	Any        =					subparsers.add_parser("""config"""      ,description=_lowerCamelCase     )
  else:
    _lowerCAmelCase						:	Union[str, Any]        =					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						:	Any     )	->				int:
  _lowerCAmelCase						:	Tuple        =					get_user_input()
  if args.config_file is not None:
    _lowerCAmelCase						:	List[str]        =					args.config_file
  else:
    if not os.path.isdir(_lowerCamelCase     ):
      os.makedirs(_lowerCamelCase     )
    _lowerCAmelCase						:	Tuple        =					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					(      )	->				Tuple:
  _lowerCAmelCase						:	Optional[Any]        =					config_command_parser()
  _lowerCAmelCase						:	List[Any]        =					parser.parse_args()
  config_command(_lowerCamelCase     )


if __name__ == "__main__":
     main()





"""simple docstring"""




import socket





def 							SCREAMING_SNAKE_CASE					(      )	->				str:
  _lowerCAmelCase						:	Optional[int]        =					socket.socket(socket.AF_INET      ,socket.SOCK_STREAM     )
  _lowerCAmelCase						:	Optional[int]        =					socket.gethostname()
  _lowerCAmelCase						:	Tuple        =					12312

  sock.connect((host, port)     )
  sock.send(b"""Hello server!"""     )

  with open("""Received_file"""      ,"""wb"""     ) as out_file:
    print("""File opened"""     )
    print("""Receiving data..."""     )
    while True:
      _lowerCAmelCase						:	List[Any]        =					sock.recv(1024     )
      if not data:
        break
      out_file.write(_lowerCamelCase     )

  print("""Successfully received the file"""     )
  sock.close()
  print("""Connection closed"""     )


if __name__ == "__main__":
     main()




