



"""simple docstring"""






from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_flax_available,
    is_tf_available,
    is_tokenizers_available,
    is_torch_available,
)


__UpperCAmelCase   :   Optional[Any]        =       {
    "configuration_whisper": ["WHISPER_PRETRAINED_CONFIG_ARCHIVE_MAP", "WhisperConfig", "WhisperOnnxConfig"],
    "feature_extraction_whisper": ["WhisperFeatureExtractor"],
    "processing_whisper": ["WhisperProcessor"],
    "tokenization_whisper": ["WhisperTokenizer"],
}

try:
				if not is_tokenizers_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				__UpperCAmelCase   :   str        =       ["WhisperTokenizerFast"]

try:
				if not is_torch_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				__UpperCAmelCase   :   Dict        =       [
				    "WHISPER_PRETRAINED_MODEL_ARCHIVE_LIST",
				    "WhisperForConditionalGeneration",
				    "WhisperModel",
				    "WhisperPreTrainedModel",
				    "WhisperForAudioClassification",
				]

try:
				if not is_tf_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				__UpperCAmelCase   :   Any        =       [
				    "TF_WHISPER_PRETRAINED_MODEL_ARCHIVE_LIST",
				    "TFWhisperForConditionalGeneration",
				    "TFWhisperModel",
				    "TFWhisperPreTrainedModel",
				]

try:
				if not is_flax_available():
								raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
				pass
else:
				__UpperCAmelCase   :   List[str]        =       [
				    "FlaxWhisperForConditionalGeneration",
				    "FlaxWhisperModel",
				    "FlaxWhisperPreTrainedModel",
				    "FlaxWhisperForAudioClassification",
				]


if TYPE_CHECKING:
				from .configuration_whisper import WHISPER_PRETRAINED_CONFIG_ARCHIVE_MAP, WhisperConfig, WhisperOnnxConfig
				from .feature_extraction_whisper import WhisperFeatureExtractor
				from .processing_whisper import WhisperProcessor
				from .tokenization_whisper import WhisperTokenizer

				try:
								if not is_tokenizers_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .tokenization_whisper_fast import WhisperTokenizerFast

				try:
								if not is_torch_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .modeling_whisper import (
								    WHISPER_PRETRAINED_MODEL_ARCHIVE_LIST,
								    WhisperForAudioClassification,
								    WhisperForConditionalGeneration,
								    WhisperModel,
								    WhisperPreTrainedModel,
								)

				try:
								if not is_tf_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .modeling_tf_whisper import (
								    TF_WHISPER_PRETRAINED_MODEL_ARCHIVE_LIST,
								    TFWhisperForConditionalGeneration,
								    TFWhisperModel,
								    TFWhisperPreTrainedModel,
								)

				try:
								if not is_flax_available():
												raise OptionalDependencyNotAvailable()
				except OptionalDependencyNotAvailable:
								pass
				else:
								from .modeling_flax_whisper import (
								    FlaxWhisperForAudioClassification,
								    FlaxWhisperForConditionalGeneration,
								    FlaxWhisperModel,
								    FlaxWhisperPreTrainedModel,
								)

else:
				import sys

				__UpperCAmelCase   :   Optional[int]        =       _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)





# 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.test_utils import execute_subprocess_async





def 	_lowerCAmelCase					( UpperCamelCase__:  List[Any]=None    )			-> Union[str, Any]:





	"""simple docstring"""







	if subparsers is not None:
		A             =					subparsers.add_parser("""test"""    )
	else:
		A             =					argparse.ArgumentParser("""Accelerate test command"""    )

	parser.add_argument(
	    """--config_file"""				,       default=UpperCamelCase__				,       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=UpperCamelCase__    )
	return parser





def 	_lowerCAmelCase					( UpperCamelCase__:  Union[str, Any]    )			-> Any:





	"""simple docstring"""







	A             =					os.path.sep.join(__file__.split(os.path.sep    )[:-2] + ["""test_utils""", """scripts""", """test_script.py"""]    )

	if args.config_file is None:
		A             =					script_name
	else:
		A             =					f'--config_file={args.config_file} {script_name}'

	A             =					["""accelerate-launch"""] + test_args.split()
	A             =					execute_subprocess_async(UpperCamelCase__				,       env=os.environ.copy()    )
	if result.returncode == 0:
		print("""Test is a success! You are ready for your distributed training!"""    )





def 	_lowerCAmelCase					( )			-> List[Any]:





	"""simple docstring"""







	A             =					test_command_parser()
	A             =					parser.parse_args()
	test_command(UpperCamelCase__    )


if __name__ == "__main__":
				main()



