'''simple docstring'''




from typing import List

import datasets
from datasets.tasks import AudioClassification

from ..folder_based_builder import folder_based_builder


_a   :	Dict 							=   datasets.utils.logging.get_logger(__name__)






class 		__A       (folder_based_builder.FolderBasedBuilderConfig   ):
       snake_case       :bool		    =					None
       snake_case       :bool		    =					None






class 		__A       (folder_based_builder.FolderBasedBuilder   ):
       snake_case       :Dict		    =					datasets.Audio()
       snake_case       :List[str]		    =					"audio"
       snake_case       :List[str]		    =					AudioFolderConfig
       snake_case       :List[str]  # definition at the bottom of the script
       snake_case       :Optional[Any]		    =					AudioClassification(audio_column="audio" ,					label_column="label"   )





_a   :	Tuple 							=   [
    ".aiff",
    ".au",
    ".avr",
    ".caf",
    ".flac",
    ".htk",
    ".svx",
    ".mat4",
    ".mat5",
    ".mpc2k",
    ".ogg",
    ".paf",
    ".pvf",
    ".raw",
    ".rf64",
    ".sd2",
    ".sds",
    ".ircam",
    ".voc",
    ".w64",
    ".wav",
    ".nist",
    ".wavex",
    ".wve",
    ".xi",
    ".mp3",
    ".opus",
]
_a   :	Tuple 							=   AUDIO_EXTENSIONS







'''simple docstring'''




from collections import OrderedDict
from typing import Mapping

from ...configuration_utils import PretrainedConfig
from ...onnx import OnnxConfig
from ...utils import logging


_a   :	List[str] 							=   logging.get_logger(__name__)

_a   :	Any 							=   {
    "kssteven/ibert-roberta-base": "https://huggingface.co/kssteven/ibert-roberta-base/resolve/main/config.json",
    "kssteven/ibert-roberta-large": "https://huggingface.co/kssteven/ibert-roberta-large/resolve/main/config.json",
    "kssteven/ibert-roberta-large-mnli": (
        "https://huggingface.co/kssteven/ibert-roberta-large-mnli/resolve/main/config.json"
    ),
}






class 		__A       (__magic_name__   ):
       snake_case       :Union[str, Any]				    =					"ibert"






       def __init__(      self				,     UpperCamelCase_=3_05_22				,     UpperCamelCase_=7_68				,     UpperCamelCase_=12				,     UpperCamelCase_=12				,     UpperCamelCase_=30_72				,     UpperCamelCase_="gelu"				,     UpperCamelCase_=0.1				,     UpperCamelCase_=0.1				,     UpperCamelCase_=5_12				,     UpperCamelCase_=2				,     UpperCamelCase_=0.0_2				,     UpperCamelCase_=1E-12				,     UpperCamelCase_=1				,     UpperCamelCase_=0				,     UpperCamelCase_=2				,     UpperCamelCase_="absolute"				,     UpperCamelCase_=False				,     UpperCamelCase_="none"				,     **UpperCamelCase_				,     ):
              super().__init__(pad_token_id=UpperCamelCase_				,     bos_token_id=UpperCamelCase_				,     eos_token_id=UpperCamelCase_				,     **UpperCamelCase_      )

              __UpperCAmelCase  :				List[Any]       				=		vocab_size
              __UpperCAmelCase  :				Optional[Any]       				=		hidden_size
              __UpperCAmelCase  :				List[Any]       				=		num_hidden_layers
              __UpperCAmelCase  :				Any       				=		num_attention_heads
              __UpperCAmelCase  :				List[str]       				=		hidden_act
              __UpperCAmelCase  :				List[str]       				=		intermediate_size
              __UpperCAmelCase  :				Optional[int]       				=		hidden_dropout_prob
              __UpperCAmelCase  :				Union[str, Any]       				=		attention_probs_dropout_prob
              __UpperCAmelCase  :				str       				=		max_position_embeddings
              __UpperCAmelCase  :				List[str]       				=		type_vocab_size
              __UpperCAmelCase  :				Dict       				=		initializer_range
              __UpperCAmelCase  :				Optional[int]       				=		layer_norm_eps
              __UpperCAmelCase  :				Any       				=		position_embedding_type
              __UpperCAmelCase  :				Tuple       				=		quant_mode
              __UpperCAmelCase  :				Union[str, Any]       				=		force_dequant






class 		__A       (__magic_name__   ):






       @property
       def 			_snake_case						(      self      ):
              if self.task == "multiple-choice":
                     __UpperCAmelCase  :				Optional[int]       				=		{0: "batch", 1: "choice", 2: "sequence"}
              else:
                     __UpperCAmelCase  :				Optional[int]       				=		{0: "batch", 1: "sequence"}
              return OrderedDict(
                  [
                      ("input_ids", dynamic_axis),
                      ("attention_mask", dynamic_axis),
                  ]      )






