
'''simple docstring'''
import copy
import os
from typing import Union

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


__lowerCAmelCase					=			logging.get_logger(__name__)

__lowerCAmelCase					=			{
    'google/pix2struct-textcaps-base': (
        'https://huggingface.co/google/pix2struct-textcaps-base/resolve/main/config.json'
    ),
}


class      _lowerCAmelCase       (       lowercase__ ):







							'''simple docstring'''

							lowerCAmelCase_   		=		"pix2struct_text_model"
							lowerCAmelCase_   		=		["past_key_values"]
							lowerCAmelCase_   		=		{
							    "hidden_size": "hidden_size",
							    "num_attention_heads": "num_heads",
							    "num_hidden_layers": "num_layers",
							}

							def __init__(self    ,    UpperCAmelCase=50244    ,    UpperCAmelCase=768    ,    UpperCAmelCase=64    ,    UpperCAmelCase=2048    ,    UpperCAmelCase=12    ,    UpperCAmelCase=12    ,    UpperCAmelCase=32    ,    UpperCAmelCase=128    ,    UpperCAmelCase=0.1    ,    UpperCAmelCase=1e-6    ,    UpperCAmelCase=1.0    ,    UpperCAmelCase="gelu_new"    ,    UpperCAmelCase=0    ,    UpperCAmelCase=False    ,    UpperCAmelCase=0    ,    UpperCAmelCase=1    ,    UpperCAmelCase=False    ,    UpperCAmelCase=True    ,    **UpperCAmelCase    ,    )	->  str:
														_snake_case  =    vocab_size
														_snake_case  =    hidden_size
														_snake_case  =    d_kv
														_snake_case  =    d_ff
														_snake_case  =    num_layers
														_snake_case  =    num_heads
														_snake_case  =    relative_attention_num_buckets
														_snake_case  =    relative_attention_max_distance
														_snake_case  =    dropout_rate
														_snake_case  =    layer_norm_epsilon
														_snake_case  =    initializer_factor
														_snake_case  =    use_cache

														_snake_case  =    eos_token_id
														_snake_case  =    decoder_start_token_id

														# for backwards compatibility
														_snake_case  =    dense_act_fn

														super().__init__(
														    pad_token_id=lowercase_    ,    eos_token_id=lowercase_    ,    decoder_start_token_id=lowercase_    ,    tie_word_embeddings=lowercase_    ,    is_decoder=lowercase_    ,    **lowercase_    ,    )





							@classmethod
							def 			lowercase				(cls    ,    UpperCAmelCase    ,    **UpperCAmelCase    )	->  "PretrainedConfig":
														cls._set_token_in_kwargs(lowercase_    )

														_snake_case,				_snake_case  =    cls.get_config_dict(lowercase_    ,    **lowercase_    )

														# get the text config dict if we are loading from Pix2StructConfig
														if config_dict.get("""model_type"""    ) == "pix2struct":
																					_snake_case  =    config_dict["""text_config"""]

														if "model_type" in config_dict and hasattr(cls    ,    """model_type"""    ) and config_dict["model_type"] != cls.model_type:
																					logger.warning(
																					    f"""You are using a model of type {config_dict["model_type"]} to instantiate a model of type """
																					    f"""{cls.model_type}. This is not supported for all configurations of models and can yield errors."""    )

														return cls.from_dict(lowercase_    ,    **lowercase_    )


class      _lowerCAmelCase       (       lowercase__ ):







							'''simple docstring'''

							lowerCAmelCase_   		=		"pix2struct_vision_model"

							def __init__(self    ,    UpperCAmelCase=768    ,    UpperCAmelCase=768    ,    UpperCAmelCase=2048    ,    UpperCAmelCase=64    ,    UpperCAmelCase=12    ,    UpperCAmelCase=12    ,    UpperCAmelCase="gelu_new"    ,    UpperCAmelCase=1e-6    ,    UpperCAmelCase=0.0    ,    UpperCAmelCase=0.0    ,    UpperCAmelCase=1e-1_0    ,    UpperCAmelCase=1.0    ,    UpperCAmelCase=4096    ,    UpperCAmelCase=32    ,    UpperCAmelCase=128    ,    **UpperCAmelCase    ,    )	->  Optional[Any]:
														super().__init__(**lowercase_    )

														_snake_case  =    hidden_size
														_snake_case  =    patch_embed_hidden_size
														_snake_case  =    d_ff
														_snake_case  =    dropout_rate
														_snake_case  =    num_hidden_layers
														_snake_case  =    num_attention_heads
														_snake_case  =    initializer_range
														_snake_case  =    initializer_factor
														_snake_case  =    attention_dropout
														_snake_case  =    layer_norm_eps
														_snake_case  =    dense_act_fn
														_snake_case  =    seq_len
														_snake_case  =    relative_attention_num_buckets
														_snake_case  =    relative_attention_max_distance
														_snake_case  =    d_kv





							@classmethod
							def 			lowercase				(cls    ,    UpperCAmelCase    ,    **UpperCAmelCase    )	->  "PretrainedConfig":
														cls._set_token_in_kwargs(lowercase_    )

														_snake_case,				_snake_case  =    cls.get_config_dict(lowercase_    ,    **lowercase_    )

														# get the vision config dict if we are loading from Pix2StructConfig
														if config_dict.get("""model_type"""    ) == "pix2struct":
																					_snake_case  =    config_dict["""vision_config"""]

														if "model_type" in config_dict and hasattr(cls    ,    """model_type"""    ) and config_dict["model_type"] != cls.model_type:
																					logger.warning(
																					    f"""You are using a model of type {config_dict["model_type"]} to instantiate a model of type """
																					    f"""{cls.model_type}. This is not supported for all configurations of models and can yield errors."""    )

														return cls.from_dict(lowercase_    ,    **lowercase_    )


class      _lowerCAmelCase       (       lowercase__ ):







							'''simple docstring'''

							lowerCAmelCase_   		=		"pix2struct"
							lowerCAmelCase_   		=		True

							def __init__(self    ,    UpperCAmelCase=None    ,    UpperCAmelCase=None    ,    UpperCAmelCase=1.0    ,    UpperCAmelCase=0.02    ,    UpperCAmelCase=False    ,    UpperCAmelCase=False    ,    UpperCAmelCase=True    ,    **UpperCAmelCase    ,    )	->  str:
														super().__init__(tie_word_embeddings=lowercase_    ,    is_encoder_decoder=lowercase_    ,    **lowercase_    )

														if text_config is None:
																					_snake_case  =    {}
																					logger.info("""text_config is None. Initializing the Pix2StructTextConfig with default values."""    )

														if vision_config is None:
																					_snake_case  =    {}
																					logger.info("""vision_config is None. Initializing the Pix2StructVisionConfig with default values."""    )

														_snake_case  =    PixaStructTextConfig(**lowercase_    )
														_snake_case  =    PixaStructVisionConfig(**lowercase_    )

														_snake_case  =    self.text_config.decoder_start_token_id
														_snake_case  =    self.text_config.pad_token_id
														_snake_case  =    self.text_config.eos_token_id

														_snake_case  =    initializer_factor
														_snake_case  =    initializer_range

														_snake_case  =    self.initializer_range
														_snake_case  =    self.initializer_range

														_snake_case  =    is_vqa

							@classmethod
							def 			lowercase				(cls    ,    UpperCAmelCase    ,    UpperCAmelCase    ,    **UpperCAmelCase    )	->  Any:
														return cls(text_config=text_config.to_dict()    ,    vision_config=vision_config.to_dict()    ,    **lowercase_    )





							def 			lowercase				(self    )	->  Union[str, Any]:
														_snake_case  =    copy.deepcopy(self.__dict__    )
														_snake_case  =    self.text_config.to_dict()
														_snake_case  =    self.vision_config.to_dict()
														_snake_case  =    self.__class__.model_type
														return output



import argparse
import json
import os
import sys
import tempfile
import unittest
from argparse import Namespace
from dataclasses import dataclass, field
from enum import Enum
from pathlib import Path
from typing import List, Literal, Optional

import yaml

from transformers import HfArgumentParser, TrainingArguments
from transformers.hf_argparser import make_choice_type_function, string_to_bool


# Since Python 3.10, we can use the builtin `|` operator for Union types
# See PEP 604: https://peps.python.org/pep-0604
_lowerCAmelCase:						Any   				=       sys.version_info >= (3, 10)






def   _lowercase(					__a				:						int=None    ,     __a				:						Any=None     ):
							return field(default_factory=lambda: default    ,     metadata=__a     )




@dataclass
class       lowercase_					:
				snake_case	    =42
				snake_case	    =42
				snake_case	    =42
				snake_case	    =42







@dataclass
class       lowercase_					:
				snake_case	    =42
				snake_case	    =field(default='toto'      ,					metadata={'help': 'help message'}   )







@dataclass
class       lowercase_					:
				snake_case	    =False
				snake_case	    =True
				snake_case	    =None







class       lowercase_					(lowercase__   ):
				snake_case	    ='titi'
				snake_case	    ='toto'







class       lowercase_					(lowercase__   ):
				snake_case	    ='titi'
				snake_case	    ='toto'
				snake_case	    =42







@dataclass
class       lowercase_					:
				snake_case	    ="toto"





				def 						__UpperCamelCase    (				self)  -> List[str]:
											a__      =BasicEnum(self.foo)







@dataclass
class       lowercase_					:
				snake_case	    ="toto"





				def 						__UpperCamelCase    (				self)  -> List[str]:
											a__      =MixedTypeEnum(self.foo)







@dataclass
class       lowercase_					:
				snake_case	    =None
				snake_case	    =field(default=lowercase__      ,					metadata={'help': 'help message'}   )
				snake_case	    =None
				snake_case	    =list_field(default=[]   )
				snake_case	    =list_field(default=[]   )







@dataclass
class       lowercase_					:
				snake_case	    =list_field(default=[]   )
				snake_case	    =list_field(default=[1, 2, 3]   )
				snake_case	    =list_field(default=['Hallo', 'Bonjour', 'Hello']   )
				snake_case	    =list_field(default=[0.1, 0.2, 0.3]   )







@dataclass
class       lowercase_					:
				snake_case	    =field()
				snake_case	    =field()
				snake_case	    =field()





				def 						__UpperCamelCase    (				self)  -> List[Any]:
											a__      =BasicEnum(self.required_enum)







@dataclass
class       lowercase_					:
				snake_case	    =42
				snake_case	    =field()
				snake_case	    =None
				snake_case	    =field(default='toto'      ,					metadata={'help': 'help message'}   )
				snake_case	    =list_field(default=['Hallo', 'Bonjour', 'Hello']   )


if is_python_no_less_than_3_10:




	@dataclass
	class       lowercase_					:
					snake_case	    =False
					snake_case	    =True
					snake_case	    =None







	@dataclass
	class       lowercase_					:
					snake_case	    =None
					snake_case	    =field(default=lowercase__      ,					metadata={'help': 'help message'}   )
					snake_case	    =None
					snake_case	    =list_field(default=[]   )
					snake_case	    =list_field(default=[]   )







class       lowercase_					(unittest.TestCase   ):





				def 						__UpperCamelCase    (				self       ,  lowercase_       ,  lowercase_)  -> int:
											self.assertEqual(len(a._actions)       ,  len(b._actions))
											for x, y in zip(a._actions       ,  b._actions):
																		a__      ={k: v for k, v in vars(lowercase_).items() if k != 'container'}
																		a__      ={k: v for k, v in vars(lowercase_).items() if k != 'container'}

																		# Choices with mixed type have custom function as "type"
																		# So we need to compare results directly for equality
																		if xx.get('choices'       ,  lowercase_) and yy.get('choices'       ,  lowercase_):
																									for expected_choice in yy["choices"] + xx["choices"]:
																																self.assertEqual(xx['type'](lowercase_)       ,  yy['type'](lowercase_))
																									del xx["type"], yy["type"]

																		self.assertEqual(lowercase_       ,  lowercase_)





				def 						__UpperCamelCase    (				self)  -> List[Any]:
											a__      =HfArgumentParser(lowercase_)

											a__      =argparse.ArgumentParser()
											expected.add_argument('--foo'       ,  type=lowercase_       ,  required=lowercase_)
											expected.add_argument('--bar'       ,  type=lowercase_       ,  required=lowercase_)
											expected.add_argument('--baz'       ,  type=lowercase_       ,  required=lowercase_)
											expected.add_argument('--flag'       ,  type=lowercase_       ,  default=lowercase_       ,  const=lowercase_       ,  nargs='?')
											self.argparsersEqual(lowercase_       ,  lowercase_)

											a__      =['--foo', '1', '--baz', 'quux', '--bar', '0.5']
											((a__)    ,     )      =parser.parse_args_into_dataclasses(lowercase_       ,  look_for_args_file=lowercase_)
											self.assertFalse(example.flag)





				def 						__UpperCamelCase    (				self)  -> Union[str, Any]:
											a__      =HfArgumentParser(lowercase_)

											a__      =argparse.ArgumentParser()
											expected.add_argument('--foo'       ,  default=42       ,  type=lowercase_)
											expected.add_argument('--baz'       ,  default='toto'       ,  type=lowercase_       ,  help='help message')
											self.argparsersEqual(lowercase_       ,  lowercase_)





				def 						__UpperCamelCase    (				self)  -> Optional[Any]:
											a__      =argparse.ArgumentParser()
											expected.add_argument('--foo'       ,  type=lowercase_       ,  default=lowercase_       ,  const=lowercase_       ,  nargs='?')
											expected.add_argument('--baz'       ,  type=lowercase_       ,  default=lowercase_       ,  const=lowercase_       ,  nargs='?')
											# A boolean no_* argument always has to come after its "default: True" regular counter-part
											# and its default must be set to False
											expected.add_argument('--no_baz'       ,  action='store_false'       ,  default=lowercase_       ,  dest='baz')
											expected.add_argument('--opt'       ,  type=lowercase_       ,  default=lowercase_)

											a__      =[WithDefaultBoolExample]
											if is_python_no_less_than_3_10:
																		dataclass_types.append(lowercase_)

											for dataclass_type in dataclass_types:
																		a__      =HfArgumentParser(lowercase_)
																		self.argparsersEqual(lowercase_       ,  lowercase_)

																		a__      =parser.parse_args([])
																		self.assertEqual(lowercase_       ,  Namespace(foo=lowercase_       ,  baz=lowercase_       ,  opt=lowercase_))

																		a__      =parser.parse_args(['--foo', '--no_baz'])
																		self.assertEqual(lowercase_       ,  Namespace(foo=lowercase_       ,  baz=lowercase_       ,  opt=lowercase_))

																		a__      =parser.parse_args(['--foo', '--baz'])
																		self.assertEqual(lowercase_       ,  Namespace(foo=lowercase_       ,  baz=lowercase_       ,  opt=lowercase_))

																		a__      =parser.parse_args(['--foo', 'True', '--baz', 'True', '--opt', 'True'])
																		self.assertEqual(lowercase_       ,  Namespace(foo=lowercase_       ,  baz=lowercase_       ,  opt=lowercase_))

																		a__      =parser.parse_args(['--foo', 'False', '--baz', 'False', '--opt', 'False'])
																		self.assertEqual(lowercase_       ,  Namespace(foo=lowercase_       ,  baz=lowercase_       ,  opt=lowercase_))





				def 						__UpperCamelCase    (				self)  -> str:
											a__      =HfArgumentParser(lowercase_)

											a__      =argparse.ArgumentParser()
											expected.add_argument(
											    '--foo'       ,  default='toto'       ,  choices=['titi', 'toto', 42]       ,  type=make_choice_type_function(['titi', 'toto', 42])       ,  )
											self.argparsersEqual(lowercase_       ,  lowercase_)

											a__      =parser.parse_args([])
											self.assertEqual(args.foo       ,  'toto')
											a__      =parser.parse_args_into_dataclasses([])[0]
											self.assertEqual(enum_ex.foo       ,  MixedTypeEnum.toto)

											a__      =parser.parse_args(['--foo', 'titi'])
											self.assertEqual(args.foo       ,  'titi')
											a__      =parser.parse_args_into_dataclasses(['--foo', 'titi'])[0]
											self.assertEqual(enum_ex.foo       ,  MixedTypeEnum.titi)

											a__      =parser.parse_args(['--foo', '42'])
											self.assertEqual(args.foo       ,  42)
											a__      =parser.parse_args_into_dataclasses(['--foo', '42'])[0]
											self.assertEqual(enum_ex.foo       ,  MixedTypeEnum.fourtytwo)





				def 						__UpperCamelCase    (				self)  -> List[Any]:




											@dataclass
											class       lowercase_					:
															snake_case	    ="toto"







											a__      =HfArgumentParser(lowercase_)

											a__      =argparse.ArgumentParser()
											expected.add_argument(
											    '--foo'       ,  default='toto'       ,  choices=('titi', 'toto', 42)       ,  type=make_choice_type_function(['titi', 'toto', 42])       ,  )
											self.argparsersEqual(lowercase_       ,  lowercase_)

											a__      =parser.parse_args([])
											self.assertEqual(args.foo       ,  'toto')

											a__      =parser.parse_args(['--foo', 'titi'])
											self.assertEqual(args.foo       ,  'titi')

											a__      =parser.parse_args(['--foo', '42'])
											self.assertEqual(args.foo       ,  42)





				def 						__UpperCamelCase    (				self)  -> Optional[int]:
											a__      =HfArgumentParser(lowercase_)

											a__      =argparse.ArgumentParser()
											expected.add_argument('--foo_int'       ,  nargs='+'       ,  default=[]       ,  type=lowercase_)
											expected.add_argument('--bar_int'       ,  nargs='+'       ,  default=[1, 2, 3]       ,  type=lowercase_)
											expected.add_argument('--foo_str'       ,  nargs='+'       ,  default=['Hallo', 'Bonjour', 'Hello']       ,  type=lowercase_)
											expected.add_argument('--foo_float'       ,  nargs='+'       ,  default=[0.1, 0.2, 0.3]       ,  type=lowercase_)

											self.argparsersEqual(lowercase_       ,  lowercase_)

											a__      =parser.parse_args([])
											self.assertEqual(
											    lowercase_       ,  Namespace(foo_int=[]       ,  bar_int=[1, 2, 3]       ,  foo_str=['Hallo', 'Bonjour', 'Hello']       ,  foo_float=[0.1, 0.2, 0.3])       ,  )

											a__      =parser.parse_args('--foo_int 1 --bar_int 2 3 --foo_str a b c --foo_float 0.1 0.7'.split())
											self.assertEqual(lowercase_       ,  Namespace(foo_int=[1]       ,  bar_int=[2, 3]       ,  foo_str=['a', 'b', 'c']       ,  foo_float=[0.1, 0.7]))





				def 						__UpperCamelCase    (				self)  -> Dict:
											a__      =argparse.ArgumentParser()
											expected.add_argument('--foo'       ,  default=lowercase_       ,  type=lowercase_)
											expected.add_argument('--bar'       ,  default=lowercase_       ,  type=lowercase_       ,  help='help message')
											expected.add_argument('--baz'       ,  default=lowercase_       ,  type=lowercase_)
											expected.add_argument('--ces'       ,  nargs='+'       ,  default=[]       ,  type=lowercase_)
											expected.add_argument('--des'       ,  nargs='+'       ,  default=[]       ,  type=lowercase_)

											a__      =[OptionalExample]
											if is_python_no_less_than_3_10:
																		dataclass_types.append(lowercase_)

											for dataclass_type in dataclass_types:
																		a__      =HfArgumentParser(lowercase_)

																		self.argparsersEqual(lowercase_       ,  lowercase_)

																		a__      =parser.parse_args([])
																		self.assertEqual(lowercase_       ,  Namespace(foo=lowercase_       ,  bar=lowercase_       ,  baz=lowercase_       ,  ces=[]       ,  des=[]))

																		a__      =parser.parse_args('--foo 12 --bar 3.14 --baz 42 --ces a b c --des 1 2 3'.split())
																		self.assertEqual(lowercase_       ,  Namespace(foo=12       ,  bar=3.14       ,  baz='42'       ,  ces=['a', 'b', 'c']       ,  des=[1, 2, 3]))





				def 						__UpperCamelCase    (				self)  -> str:
											a__      =HfArgumentParser(lowercase_)

											a__      =argparse.ArgumentParser()
											expected.add_argument('--required_list'       ,  nargs='+'       ,  type=lowercase_       ,  required=lowercase_)
											expected.add_argument('--required_str'       ,  type=lowercase_       ,  required=lowercase_)
											expected.add_argument(
											    '--required_enum'       ,  type=make_choice_type_function(['titi', 'toto'])       ,  choices=['titi', 'toto']       ,  required=lowercase_       ,  )
											self.argparsersEqual(lowercase_       ,  lowercase_)





				def 						__UpperCamelCase    (				self)  -> str:
											a__      =HfArgumentParser(lowercase_)

											a__      =argparse.ArgumentParser()
											expected.add_argument('--foo'       ,  type=lowercase_       ,  required=lowercase_)
											expected.add_argument(
											    '--required_enum'       ,  type=make_choice_type_function(['titi', 'toto'])       ,  choices=['titi', 'toto']       ,  required=lowercase_       ,  )
											expected.add_argument('--opt'       ,  type=lowercase_       ,  default=lowercase_)
											expected.add_argument('--baz'       ,  default='toto'       ,  type=lowercase_       ,  help='help message')
											expected.add_argument('--foo_str'       ,  nargs='+'       ,  default=['Hallo', 'Bonjour', 'Hello']       ,  type=lowercase_)
											self.argparsersEqual(lowercase_       ,  lowercase_)





				def 						__UpperCamelCase    (				self)  -> List[Any]:
											a__      =HfArgumentParser(lowercase_)

											a__      ={
											    'foo': 12,
											    'bar': 3.14,
											    'baz': '42',
											    'flag': True,
											}

											a__      =parser.parse_dict(lowercase_)[0]
											a__      =BasicExample(**lowercase_)
											self.assertEqual(lowercase_       ,  lowercase_)





				def 						__UpperCamelCase    (				self)  -> Union[str, Any]:
											a__      =HfArgumentParser(lowercase_)

											a__      ={
											    'foo': 12,
											    'bar': 3.14,
											    'baz': '42',
											    'flag': True,
											    'extra': 42,
											}

											self.assertRaises(lowercase_       ,  parser.parse_dict       ,  lowercase_       ,  allow_extra_keys=lowercase_)





				def 						__UpperCamelCase    (				self)  -> List[Any]:
											a__      =HfArgumentParser(lowercase_)

											a__      ={
											    'foo': 12,
											    'bar': 3.14,
											    'baz': '42',
											    'flag': True,
											}
											with tempfile.TemporaryDirectory() as tmp_dir:
																		a__      =os.path.join(lowercase_       ,  'temp_json')
																		os.mkdir(lowercase_)
																		with open(temp_local_path + '.json'       ,  'w+') as f:
																									json.dump(lowercase_       ,  lowercase_)
																		a__      =parser.parse_yaml_file(Path(temp_local_path + '.json'))[0]

											a__      =BasicExample(**lowercase_)
											self.assertEqual(lowercase_       ,  lowercase_)





				def 						__UpperCamelCase    (				self)  -> Any:
											a__      =HfArgumentParser(lowercase_)

											a__      ={
											    'foo': 12,
											    'bar': 3.14,
											    'baz': '42',
											    'flag': True,
											}
											with tempfile.TemporaryDirectory() as tmp_dir:
																		a__      =os.path.join(lowercase_       ,  'temp_yaml')
																		os.mkdir(lowercase_)
																		with open(temp_local_path + '.yaml'       ,  'w+') as f:
																									yaml.dump(lowercase_       ,  lowercase_)
																		a__      =parser.parse_yaml_file(Path(temp_local_path + '.yaml'))[0]
											a__      =BasicExample(**lowercase_)
											self.assertEqual(lowercase_       ,  lowercase_)





				def 						__UpperCamelCase    (				self)  -> Union[str, Any]:
											a__      =HfArgumentParser(lowercase_)
											self.assertIsNotNone(lowercase_)

