





'''simple docstring'''






from typing import Callable, Optional

from .. import Features
from ..packaged_modules.generator.generator import Generator
from .abc import AbstractDatasetInputStream




class 					UpperCAmelCase	(     snake_case_  ):




      def __init__(     self     :					Union[str, Any]     ,     __snake_case     :					Callable     ,     __snake_case     :					Optional[Features] = None     ,     __snake_case     :					str = None     ,     __snake_case     :					bool = False     ,     __snake_case     :					bool = False     ,     __snake_case     :					Optional[dict] = None     ,     __snake_case     :					Optional[int] = None     ,     **__snake_case     :					Optional[Any]     ,     )							-> Union[str, Any]:
            super().__init__(
                features=__snake_case     ,     cache_dir=__snake_case     ,     keep_in_memory=__snake_case     ,     streaming=__snake_case     ,     num_proc=__snake_case     ,     **__snake_case     ,     )
            _lowerCAmelCase					=				Generator(
                cache_dir=__snake_case     ,     features=__snake_case     ,     generator=__snake_case     ,     gen_kwargs=__snake_case     ,     **__snake_case     ,     )




      def 							lowercase__       (     self     :					str    )							-> Dict:
            # Build iterable dataset
            if self.streaming:
                  _lowerCAmelCase					=				self.builder.as_streaming_dataset(split="""train"""    )
            # Build regular (map-style) dataset
            else:
                  _lowerCAmelCase					=				None
                  _lowerCAmelCase					=				None
                  _lowerCAmelCase					=				None
                  _lowerCAmelCase					=				None

                  self.builder.download_and_prepare(
                      download_config=__snake_case     ,     download_mode=__snake_case     ,     verification_mode=__snake_case     ,     base_path=__snake_case     ,     num_proc=self.num_proc     ,     )
                  _lowerCAmelCase					=				self.builder.as_dataset(
                      split="""train"""     ,     verification_mode=__snake_case     ,     in_memory=self.keep_in_memory    )
            return dataset













'''simple docstring'''






from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_sentencepiece_available,
    is_torch_available,
)


A__							:  Any    ={
    '''configuration_speecht5''': [
        '''SPEECHT5_PRETRAINED_CONFIG_ARCHIVE_MAP''',
        '''SPEECHT5_PRETRAINED_HIFIGAN_CONFIG_ARCHIVE_MAP''',
        '''SpeechT5Config''',
        '''SpeechT5HifiGanConfig''',
    ],
    '''feature_extraction_speecht5''': ['''SpeechT5FeatureExtractor'''],
    '''processing_speecht5''': ['''SpeechT5Processor'''],
}

try:
      if not is_sentencepiece_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      A__							:  Union[str, Any]    =['''SpeechT5Tokenizer''']

try:
      if not is_torch_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      A__							:  List[Any]    =[
          '''SPEECHT5_PRETRAINED_MODEL_ARCHIVE_LIST''',
          '''SpeechT5ForSpeechToText''',
          '''SpeechT5ForSpeechToSpeech''',
          '''SpeechT5ForTextToSpeech''',
          '''SpeechT5Model''',
          '''SpeechT5PreTrainedModel''',
          '''SpeechT5HifiGan''',
      ]

if TYPE_CHECKING:
      from .configuration_speechta import (
          SPEECHT5_PRETRAINED_CONFIG_ARCHIVE_MAP,
          SPEECHT5_PRETRAINED_HIFIGAN_CONFIG_ARCHIVE_MAP,
          SpeechTaConfig,
          SpeechTaHifiGanConfig,
      )
      from .feature_extraction_speechta import SpeechTaFeatureExtractor
      from .processing_speechta import SpeechTaProcessor

      try:
            if not is_sentencepiece_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .tokenization_speechta import SpeechTaTokenizer

      try:
            if not is_torch_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_speechta import (
                SPEECHT5_PRETRAINED_MODEL_ARCHIVE_LIST,
                SpeechTaForSpeechToSpeech,
                SpeechTaForSpeechToText,
                SpeechTaForTextToSpeech,
                SpeechTaHifiGan,
                SpeechTaModel,
                SpeechTaPreTrainedModel,
            )

else:
      import sys

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






