


"""simple docstring"""
import argparse
import os
import subprocess

from packaging.version import Version, parse

from accelerate.commands.config.config_args import default_config_file, load_config_from_file


__SCREAMING_SNAKE_CASE							:			Optional[Any]						=			'Run commands across TPU VMs for initial setup before running `accelerate launch`.'







def   _a (					_SCREAMING_SNAKE_CASE=None )			-> Dict:



							if subparsers is not None:
														snake_case_							=						subparsers.add_parser("""tpu-config""" ,       description=_description )
							else:
														snake_case_							=						argparse.ArgumentParser("""Accelerate tpu-config command""" ,       description=_description )
							# Core arguments
							snake_case_							=						parser.add_argument_group(
							    """Config Arguments""" ,       """Arguments that can be configured through `accelerate config`.""" )
							config_args.add_argument(
							    """--config_file""" ,       type=A__ ,       default=A__ ,       help="""Path to the config file to use for accelerate.""" ,       )
							config_args.add_argument(
							    """--tpu_name""" ,       default=A__ ,       help="""The name of the TPU to use. If not specified, will use the TPU specified in the config file.""" ,       )
							config_args.add_argument(
							    """--tpu_zone""" ,       default=A__ ,       help="""The zone of the TPU to use. If not specified, will use the zone specified in the config file.""" ,       )
							snake_case_							=						parser.add_argument_group("""TPU Arguments""" ,       """Arguments for options ran inside the TPU.""" )
							pod_args.add_argument(
							    """--use_alpha""" ,       action="""store_true""" ,       help="""Whether to use `gcloud alpha` when running the TPU training script instead of `gcloud`.""" ,       )
							pod_args.add_argument(
							    """--command_file""" ,       default=A__ ,       help="""The path to the file containing the commands to run on the pod on startup.""" ,       )
							pod_args.add_argument(
							    """--command""" ,       action="""append""" ,       nargs="""+""" ,       help="""A command to run on the pod. Can be passed multiple times.""" ,       )
							pod_args.add_argument(
							    """--install_accelerate""" ,       action="""store_true""" ,       help="""Whether to install accelerate on the pod. Defaults to False.""" ,       )
							pod_args.add_argument(
							    """--accelerate_version""" ,       default="""latest""" ,       help="""The version of accelerate to install on the pod. If not specified, will use the latest pypi version. Specify \'dev\' to install from GitHub.""" ,       )
							pod_args.add_argument(
							    """--debug""" ,       action="""store_true""" ,       help="""If set, will print the command that would be run instead of running it.""" )

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







def   _a (					_SCREAMING_SNAKE_CASE )			-> Any:
							snake_case_							=						None

							# Get the default from the config file if it exists.
							if args.config_file is not None or os.path.isfile(A__ ):
														snake_case_							=						load_config_from_file(args.config_file )
														if not args.command_file and defaults.command_file is not None and not args.command:
																					snake_case_							=						defaults.command_file
														if not args.command and defaults.commands is not None:
																					snake_case_							=						defaults.commands
														if not args.tpu_name:
																					snake_case_							=						defaults.tpu_name
														if not args.tpu_zone:
																					snake_case_							=						defaults.tpu_zone
							if args.accelerate_version == "dev":
														snake_case_							=						"""git+https://github.com/huggingface/accelerate.git"""
							elif args.accelerate_version == "latest":
														snake_case_							=						"""accelerate -U"""
							elif isinstance(parse(args.accelerate_version ) ,       A__ ):
														snake_case_							=						f"""accelerate=={args.accelerate_version}"""

							if not args.command_file and not args.command:
														raise ValueError("""You must specify either a command file or a command to run on the pod.""" )

							if args.command_file:
														with open(args.command_file ,       """r""" ) as f:
																					snake_case_							=						[f.read().splitlines()]

																					 # To turn list of lists into list of strings
							if isinstance(args.command[0] ,       A__ ):
														snake_case_							=						[line for cmd in args.command for line in cmd]
							# Default to the shared folder and install accelerate
							snake_case_							=						["""cd /usr/share"""]
							if args.install_accelerate:
														new_cmd += [f"""pip install {args.accelerate_version}"""]
							new_cmd += args.command
							snake_case_							=						"""; """.join(A__ )

							# Then send it to gcloud
							# Eventually try to use google-api-core to do this instead of subprocess
							snake_case_							=						["""gcloud"""]
							if args.use_alpha:
														cmd += ["alpha"]
							cmd += [
							    "compute",
							    "tpus",
							    "tpu-vm",
							    "ssh",
							    args.tpu_name,
							    "--zone",
							    args.tpu_zone,
							    "--command",
							    args.command,
							    "--worker",
							    "all",
							]
							if args.debug:
														print(f"""Running {" ".join(A__ )}""" )
														return
							subprocess.run(A__ )
							print("""Successfully setup pod.""" )







def   _a (					)			-> Tuple:
							snake_case_							=						tpu_command_parser()
							snake_case_							=						parser.parse_args()

							tpu_command_launcher(A__ )






'''simple docstring'''






import itertools
import os
import random
import tempfile
import unittest

import numpy as np
from datasets import load_dataset

from transformers import is_speech_available
from transformers.testing_utils import check_json_file_has_correct_format, require_torch, require_torchaudio
from transformers.utils.import_utils import is_torch_available

from ...test_sequence_feature_extraction_common import SequenceFeatureExtractionTestMixin


if is_speech_available():
 from transformers import WhisperFeatureExtractor

if is_torch_available():
 import torch

lowerCAmelCase__											=	random.Random()


def 		_A     (    A__     ,				A__=1.0     ,				A__=None     ,				A__=None      ):






 """simple docstring"""



 if rng is None:
  __lowercase							  = global_rng

 __lowercase							  = []
 for batch_idx in range(shape[0]      ):
  values.append([]      )
  for _ in range(shape[1]      ):
   values[-1].append(rng.random() * scale      )

 return values






@require_torch
@require_torchaudio
class 			lowercase_	(unittest.TestCase      ):




  """simple docstring"""




  def __init__(      self   :     List[str]					,lowercase__   :     List[str]					,lowercase__   :     List[Any]=7					,lowercase__   :     Dict=4_0_0					,lowercase__   :     Tuple=2_0_0_0					,lowercase__   :     Optional[int]=1_0					,lowercase__   :     Optional[int]=1_6_0					,lowercase__   :     Dict=8					,lowercase__   :     str=0.0					,lowercase__   :     Union[str, Any]=4_0_0_0					,lowercase__   :     Optional[int]=False					,lowercase__   :     List[str]=True					,):
   __lowercase							  = parent
   __lowercase							  = batch_size
   __lowercase							  = min_seq_length
   __lowercase							  = max_seq_length
   __lowercase							  = (self.max_seq_length - self.min_seq_length) // (self.batch_size - 1)
   __lowercase							  = padding_value
   __lowercase							  = sampling_rate
   __lowercase							  = return_attention_mask
   __lowercase							  = do_normalize
   __lowercase							  = feature_size
   __lowercase							  = chunk_length
   __lowercase							  = hop_length






  def 					SCREAMING_SNAKE_CASE      (      self   :     Optional[Any]			):
   return {
       "feature_size": self.feature_size,
       "hop_length": self.hop_length,
       "chunk_length": self.chunk_length,
       "padding_value": self.padding_value,
       "sampling_rate": self.sampling_rate,
       "return_attention_mask": self.return_attention_mask,
       "do_normalize": self.do_normalize,
   }






  def 					SCREAMING_SNAKE_CASE      (      self   :     int					,lowercase__   :     List[str]=False					,lowercase__   :     Dict=False			):
   def _flatten(lowercase__   :     Optional[Any]			):
    return list(itertools.chain(*lowercase__			)			)

   if equal_length:
    __lowercase							  = [floats_list((self.max_seq_length, self.feature_size)			) for _ in range(self.batch_size			)]
   else:
    # make sure that inputs increase in size
    __lowercase							  = [
        floats_list((x, self.feature_size)			)
        for x in range(self.min_seq_length					,self.max_seq_length					,self.seq_length_diff			)
    ]
   if numpify:
    __lowercase							  = [np.asarray(lowercase__			) for x in speech_inputs]
   return speech_inputs






@require_torch
@require_torchaudio
class 			lowercase_	(lowerCamelCase__				,       unittest.TestCase      ):




  """simple docstring"""




  SCREAMING_SNAKE_CASE    :  Optional[int]							=			WhisperFeatureExtractor if is_speech_available() else None






  def 					SCREAMING_SNAKE_CASE      (      self   :     int			):
   __lowercase							  = WhisperFeatureExtractionTester(self			)






  def 					SCREAMING_SNAKE_CASE      (      self   :     Optional[Any]			):
   __lowercase							  = self.feature_extraction_class(**self.feat_extract_dict			)

   with tempfile.TemporaryDirectory() as tmpdirname:
    __lowercase							  = feat_extract_first.save_pretrained(lowercase__			)[0]
    check_json_file_has_correct_format(lowercase__			)
    __lowercase							  = self.feature_extraction_class.from_pretrained(lowercase__			)

   __lowercase							  = feat_extract_first.to_dict()
   __lowercase							  = feat_extract_second.to_dict()
   __lowercase							  = feat_extract_first.mel_filters
   __lowercase							  = feat_extract_second.mel_filters
   self.assertTrue(np.allclose(lowercase__					,lowercase__			)			)
   self.assertEqual(lowercase__					,lowercase__			)






  def 					SCREAMING_SNAKE_CASE      (      self   :     Tuple			):
   __lowercase							  = self.feature_extraction_class(**self.feat_extract_dict			)

   with tempfile.TemporaryDirectory() as tmpdirname:
    __lowercase							  = os.path.join(lowercase__					,'''feat_extract.json'''			)
    feat_extract_first.to_json_file(lowercase__			)
    __lowercase							  = self.feature_extraction_class.from_json_file(lowercase__			)

   __lowercase							  = feat_extract_first.to_dict()
   __lowercase							  = feat_extract_second.to_dict()
   __lowercase							  = feat_extract_first.mel_filters
   __lowercase							  = feat_extract_second.mel_filters
   self.assertTrue(np.allclose(lowercase__					,lowercase__			)			)
   self.assertEqual(lowercase__					,lowercase__			)






  def 					SCREAMING_SNAKE_CASE      (      self   :     List[str]			):
   # Tests that all call wrap to encode_plus and batch_encode_plus
   __lowercase							  = self.feature_extraction_class(**self.feat_extract_tester.prepare_feat_extract_dict()			)
   # create three inputs of length 800, 1000, and 1200
   __lowercase							  = [floats_list((1, x)			)[0] for x in range(8_0_0					,1_4_0_0					,2_0_0			)]
   __lowercase							  = [np.asarray(lowercase__			) for speech_input in speech_inputs]

   # Test feature size
   __lowercase							  = feature_extractor(lowercase__					,padding='''max_length'''					,return_tensors='''np'''			).input_features
   self.assertTrue(input_features.ndim == 3			)
   self.assertTrue(input_features.shape[-1] == feature_extractor.nb_max_frames			)
   self.assertTrue(input_features.shape[-2] == feature_extractor.feature_size			)

   # Test not batched input
   __lowercase							  = feature_extractor(speech_inputs[0]					,return_tensors='''np'''			).input_features
   __lowercase							  = feature_extractor(np_speech_inputs[0]					,return_tensors='''np'''			).input_features
   self.assertTrue(np.allclose(lowercase__					,lowercase__					,atol=1e-3			)			)

   # Test batched
   __lowercase							  = feature_extractor(lowercase__					,return_tensors='''np'''			).input_features
   __lowercase							  = feature_extractor(lowercase__					,return_tensors='''np'''			).input_features
   for enc_seq_a, enc_seq_a in zip(lowercase__					,lowercase__			):
    self.assertTrue(np.allclose(lowercase__					,lowercase__					,atol=1e-3			)			)

   # Test 2-D numpy arrays are batched.
   __lowercase							  = [floats_list((1, x)			)[0] for x in (8_0_0, 8_0_0, 8_0_0)]
   __lowercase							  = np.asarray(lowercase__			)
   __lowercase							  = feature_extractor(lowercase__					,return_tensors='''np'''			).input_features
   __lowercase							  = feature_extractor(lowercase__					,return_tensors='''np'''			).input_features
   for enc_seq_a, enc_seq_a in zip(lowercase__					,lowercase__			):
    self.assertTrue(np.allclose(lowercase__					,lowercase__					,atol=1e-3			)			)

   # Test truncation required
   __lowercase							  = [floats_list((1, x)			)[0] for x in range(2_0_0					,(feature_extractor.n_samples + 5_0_0)					,2_0_0			)]
   __lowercase							  = [np.asarray(lowercase__			) for speech_input in speech_inputs]

   __lowercase							  = [x[: feature_extractor.n_samples] for x in speech_inputs]
   __lowercase							  = [np.asarray(lowercase__			) for speech_input in speech_inputs_truncated]

   __lowercase							  = feature_extractor(lowercase__					,return_tensors='''np'''			).input_features
   __lowercase							  = feature_extractor(lowercase__					,return_tensors='''np'''			).input_features
   for enc_seq_a, enc_seq_a in zip(lowercase__					,lowercase__			):
    self.assertTrue(np.allclose(lowercase__					,lowercase__					,atol=1e-3			)			)






  def 					SCREAMING_SNAKE_CASE      (      self   :     Dict			):
   import torch

   __lowercase							  = self.feature_extraction_class(**self.feat_extract_tester.prepare_feat_extract_dict()			)
   __lowercase							  = np.random.rand(1_0_0					,3_2			).astype(np.floataa			)
   __lowercase							  = np_speech_inputs.tolist()

   for inputs in [py_speech_inputs, np_speech_inputs]:
    __lowercase							  = feature_extractor.pad([{'''input_features''': inputs}]					,return_tensors='''np'''			)
    self.assertTrue(np_processed.input_features.dtype == np.floataa			)
    __lowercase							  = feature_extractor.pad([{'''input_features''': inputs}]					,return_tensors='''pt'''			)
    self.assertTrue(pt_processed.input_features.dtype == torch.floataa			)






  def 					SCREAMING_SNAKE_CASE      (      self   :     Any					,lowercase__   :     int			):
   __lowercase							  = load_dataset('''hf-internal-testing/librispeech_asr_dummy'''					,'''clean'''					,split='''validation'''			)
   # automatic decoding with librispeech
   __lowercase							  = ds.sort('''id'''			).select(range(lowercase__			)			)[:num_samples]['''audio''']

   return [x["array"] for x in speech_samples]






  def 					SCREAMING_SNAKE_CASE      (      self   :     str			):
   # fmt: off
   __lowercase							  = torch.tensor(
       [
           0.1_1_9_3, -0.0_9_4_6, -0.1_0_9_8, -0.0_1_9_6, 0.0_2_2_5, -0.0_6_9_0, -0.1_7_3_6, 0.0_9_5_1,
           0.0_9_7_1, -0.0_8_1_7, -0.0_7_0_2, 0.0_1_6_2, 0.0_2_6_0, 0.0_0_1_7, -0.0_1_9_2, -0.1_6_7_8,
           0.0_7_0_9, -0.1_8_6_7, -0.0_6_5_5, -0.0_2_7_4, -0.0_2_3_4, -0.1_8_8_4, -0.0_5_1_6, -0.0_5_5_4,
           -0.0_2_7_4, -0.1_4_2_5, -0.1_4_2_3, 0.0_8_3_7, 0.0_3_7_7, -0.0_8_5_4
       ]			)
   # fmt: on

   __lowercase							  = self._load_datasamples(1			)
   __lowercase							  = WhisperFeatureExtractor()
   __lowercase							  = feature_extractor(lowercase__					,return_tensors='''pt'''			).input_features
   self.assertEqual(input_features.shape					,(1, 8_0, 3_0_0_0)			)
   self.assertTrue(torch.allclose(input_features[0, 0, :3_0]					,lowercase__					,atol=1e-4			)			)






  def 					SCREAMING_SNAKE_CASE      (      self   :     Optional[Any]			):
   __lowercase							  = self.feature_extraction_class(**self.feat_extract_tester.prepare_feat_extract_dict()			)
   __lowercase							  = self._load_datasamples(1			)[0]
   __lowercase							  = ((audio - audio.min()) / (audio.max() - audio.min())) * 6_5_5_3_5  # Rescale to [0, 65535] to show issue
   __lowercase							  = feat_extract.zero_mean_unit_var_norm([audio]					,attention_mask=lowercase__			)[0]

   self.assertTrue(np.all(np.mean(lowercase__			) < 1e-3			)			)
   self.assertTrue(np.all(np.abs(np.var(lowercase__			) - 1			) < 1e-3			)			)

