'''simple docstring'''




from typing import Optional

from ..utils.logging import get_logger
from .audio_classificiation import AudioClassification
from .automatic_speech_recognition import AutomaticSpeechRecognition
from .base import TaskTemplate
from .image_classification import ImageClassification
from .language_modeling import LanguageModeling
from .question_answering import QuestionAnsweringExtractive
from .summarization import Summarization
from .text_classification import TextClassification


__snake_case   :      List[str]						    =  [
    'AutomaticSpeechRecognition',
    'AudioClassification',
    'ImageClassification',
    'LanguageModeling',
    'QuestionAnsweringExtractive',
    'Summarization',
    'TaskTemplate',
    'TextClassification',
]

__snake_case   :      Any						    =  get_logger(__name__)


__snake_case   :      Tuple						    =  {
    AutomaticSpeechRecognition.task: AutomaticSpeechRecognition,
    AudioClassification.task: AudioClassification,
    ImageClassification.task: ImageClassification,
    LanguageModeling.task: LanguageModeling,
    QuestionAnsweringExtractive.task: QuestionAnsweringExtractive,
    Summarization.task: Summarization,
    TextClassification.task: TextClassification,
}





def      _UpperCAmelCase    (							_UpperCamelCase    :	dict   )	->      Optional[TaskTemplate]:
    A_     =   task_template_dict.get('''task'''   )
    if task_name is None:
        logger.warning(F'''Couldn\'t find template for task \'{task_name}\'. Available templates: {list(_UpperCamelCase   )}'''   )
        return None
    A_     =   NAMEaTEMPLATE.get(_UpperCamelCase   )
    return template.from_dict(_UpperCamelCase   )







'''simple docstring'''




from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_sentencepiece_available, is_tokenizers_available


__snake_case   :      Optional[int]						    =  {}

try:
  if not is_sentencepiece_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __snake_case   :      Optional[Any]						    =  ['MBart50Tokenizer']

try:
  if not is_tokenizers_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __snake_case   :      Dict						    =  ['MBart50TokenizerFast']


if TYPE_CHECKING:
  try:
    if not is_sentencepiece_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .tokenization_mbartaa import MBartaaTokenizer

  try:
    if not is_tokenizers_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .tokenization_mbartaa_fast import MBartaaTokenizerFast

else:
  import sys

  __snake_case   :      Tuple						    =  _LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)






