
from ...configuration_utils import PretrainedConfig
from ...utils import logging


lowerCamelCase_						     =					logging.get_logger(__name__)

lowerCamelCase_						     =					{
    '''sayakpaul/vit-msn-base''': '''https://huggingface.co/sayakpaul/vit-msn-base/resolve/main/config.json''',
    # See all ViT MSN models at https://huggingface.co/models?filter=vit_msn
}




class      __A(    __lowerCamelCase			):

  """simple docstring"""
  SCREAMING_SNAKE_CASE__         =      """vit_msn"""







  def __init__(self	,      SCREAMING_SNAKE_CASE_=7_68	,      SCREAMING_SNAKE_CASE_=12	,      SCREAMING_SNAKE_CASE_=12	,      SCREAMING_SNAKE_CASE_=30_72	,      SCREAMING_SNAKE_CASE_="gelu"	,      SCREAMING_SNAKE_CASE_=0.0	,      SCREAMING_SNAKE_CASE_=0.0	,      SCREAMING_SNAKE_CASE_=0.02	,      SCREAMING_SNAKE_CASE_=1E-06	,      SCREAMING_SNAKE_CASE_=2_24	,      SCREAMING_SNAKE_CASE_=16	,      SCREAMING_SNAKE_CASE_=3	,      SCREAMING_SNAKE_CASE_=True	,      **SCREAMING_SNAKE_CASE_	,      ):
       super().__init__(**SCREAMING_SNAKE_CASE_						)

       UpperCamelCase__				=    hidden_size
       UpperCamelCase__				=    num_hidden_layers
       UpperCamelCase__				=    num_attention_heads
       UpperCamelCase__				=    intermediate_size
       UpperCamelCase__				=    hidden_act
       UpperCamelCase__				=    hidden_dropout_prob
       UpperCamelCase__				=    attention_probs_dropout_prob
       UpperCamelCase__				=    initializer_range
       UpperCamelCase__				=    layer_norm_eps
       UpperCamelCase__				=    image_size
       UpperCamelCase__				=    patch_size
       UpperCamelCase__				=    num_channels
       UpperCamelCase__				=    qkv_bias







import pytest

from datasets.parallel import ParallelBackendConfig, parallel_backend
from datasets.utils.py_utils import map_nested

from .utils import require_dill_gt_0_3_2, require_joblibspark, require_not_windows






def 						__magic_name__			(   __a      :	Any						):  # picklable for multiprocessing



     '''simple docstring'''




     return i + 1






@require_dill_gt_0_3_2
@require_joblibspark
@require_not_windows
def 						__magic_name__			(   ):



     '''simple docstring'''




     with parallel_backend("""spark"""						):
          assert ParallelBackendConfig.backend_name == "spark"

     UpperCamelCase__				=    [1, 2, 3]
     with pytest.raises(__a						):
          with parallel_backend("""unsupported backend"""						):
               map_nested(__a	,		__a	,		num_proc=2						)

     with pytest.raises(__a						):
          with parallel_backend("""unsupported backend"""						):
               map_nested(__a	,		__a	,		num_proc=-1						)






@require_dill_gt_0_3_2
@require_joblibspark
@require_not_windows
@pytest.mark.parametrize("""num_proc"""	,		[2, -1]						)
def 						__magic_name__			(   __a      :	Optional[int]						):



     '''simple docstring'''




     UpperCamelCase__				=    [1, 2]
     UpperCamelCase__				=    {"""a""": 1, """b""": 2}
     UpperCamelCase__				=    {"""a""": [1, 2], """b""": [3, 4]}
     UpperCamelCase__				=    {"""a""": {"""1""": 1}, """b""": 2}
     UpperCamelCase__				=    {"""a""": 1, """b""": 2, """c""": 3, """d""": 4}
     UpperCamelCase__				=    [2, 3]
     UpperCamelCase__				=    {"""a""": 2, """b""": 3}
     UpperCamelCase__				=    {"""a""": [2, 3], """b""": [4, 5]}
     UpperCamelCase__				=    {"""a""": {"""1""": 2}, """b""": 3}
     UpperCamelCase__				=    {"""a""": 2, """b""": 3, """c""": 4, """d""": 5}

     with parallel_backend("""spark"""						):
          assert map_nested(__a	,		__a	,		num_proc=__a						) == expected_map_nested_sa
          assert map_nested(__a	,		__a	,		num_proc=__a						) == expected_map_nested_sa
          assert map_nested(__a	,		__a	,		num_proc=__a						) == expected_map_nested_sa
          assert map_nested(__a	,		__a	,		num_proc=__a						) == expected_map_nested_sa
          assert map_nested(__a	,		__a	,		num_proc=__a						) == expected_map_nested_sa





