


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
snake_case_						:			Optional[int]													=							sys.version_info >= (3, 10)







def 						lowerCamelCase(   a__=None	,a__=None):
				return field(default_factory=lambda: default	,metadata=a__)

@dataclass
class 						A__							:
						UpperCAmelCase             =						42
						UpperCAmelCase             =						42
						UpperCAmelCase             =						42
						UpperCAmelCase             =						42

@dataclass
class 						A__							:
						UpperCAmelCase             =						42
						UpperCAmelCase             =						field(default="toto"   ,     metadata={"help": "help message"}  )

@dataclass
class 						A__							:
						UpperCAmelCase             =						False
						UpperCAmelCase             =						True
						UpperCAmelCase             =						None

class 						A__							(		UpperCamelCase__  ):
						UpperCAmelCase             =						"titi"
						UpperCAmelCase             =						"toto"

class 						A__							(		UpperCamelCase__  ):
						UpperCAmelCase             =						"titi"
						UpperCAmelCase             =						"toto"
						UpperCAmelCase             =						42

@dataclass
class 						A__							:
						UpperCAmelCase             =						"toto"




						def 							__UpperCamelCase				(       self					:      Optional[Any]  )							->     Dict:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =BasicEnum(self.foo  )

@dataclass
class 						A__							:
						UpperCAmelCase             =						"toto"




						def 							__UpperCamelCase				(       self					:      int  )							->     Dict:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =MixedTypeEnum(self.foo  )

@dataclass
class 						A__							:
						UpperCAmelCase             =						None
						UpperCAmelCase             =						field(default=UpperCamelCase__   ,     metadata={"help": "help message"}  )
						UpperCAmelCase             =						None
						UpperCAmelCase             =						list_field(default=[]  )
						UpperCAmelCase             =						list_field(default=[]  )

@dataclass
class 						A__							:
						UpperCAmelCase             =						list_field(default=[]  )
						UpperCAmelCase             =						list_field(default=[1, 2, 3]  )
						UpperCAmelCase             =						list_field(default=["Hallo", "Bonjour", "Hello"]  )
						UpperCAmelCase             =						list_field(default=[0.1, 0.2, 0.3]  )

@dataclass
class 						A__							:
						UpperCAmelCase             =						field()
						UpperCAmelCase             =						field()
						UpperCAmelCase             =						field()




						def 							__UpperCamelCase				(       self					:      Optional[int]  )							->     Optional[Any]:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =BasicEnum(self.required_enum  )

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


if is_python_no_less_than_3_10:

							@dataclass
							class 						A__							:
													UpperCAmelCase             =						False
													UpperCAmelCase             =						True
													UpperCAmelCase             =						None

							@dataclass
							class 						A__							:
													UpperCAmelCase             =						None
													UpperCAmelCase             =						field(default=UpperCamelCase__   ,     metadata={"help": "help message"}  )
													UpperCAmelCase             =						None
													UpperCAmelCase             =						list_field(default=[]  )
													UpperCAmelCase             =						list_field(default=[]  )


class 						A__							(		unittest.TestCase  ):




						def 							__UpperCamelCase				(       self					:      Optional[int]				,    _a					:      argparse.ArgumentParser				,    _a					:      argparse.ArgumentParser  )							->     Dict:


										"""simple docstring"""





										self.assertEqual(len(a._actions  )				,    len(b._actions  )  )
										for x, y in zip(a._actions				,    b._actions  ):
														_SCREAMING_SNAKE_CASE			  ={k: v for k, v in vars(_a  ).items() if k != '''container'''}
														_SCREAMING_SNAKE_CASE			  ={k: v for k, v in vars(_a  ).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'''				,    _a  ) and yy.get('''choices'''				,    _a  ):
																		for expected_choice in yy["choices"] + xx["choices"]:
																						self.assertEqual(xx['''type'''](_a  )				,    yy['''type'''](_a  )  )
																		del xx["type"], yy["type"]

														self.assertEqual(_a				,    _a  )




						def 							__UpperCamelCase				(       self					:      Dict  )							->     Optional[int]:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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

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




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


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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




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


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =argparse.ArgumentParser()
										expected.add_argument('''--foo'''				,    type=_a				,    default=_a				,    const=_a				,    nargs='''?'''  )
										expected.add_argument('''--baz'''				,    type=_a				,    default=_a				,    const=_a				,    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=_a				,    dest='''baz'''  )
										expected.add_argument('''--opt'''				,    type=_a				,    default=_a  )

										_SCREAMING_SNAKE_CASE			  =[WithDefaultBoolExample]
										if is_python_no_less_than_3_10:
														dataclass_types.append(_a  )

										for dataclass_type in dataclass_types:
														_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )
														self.argparsersEqual(_a				,    _a  )

														_SCREAMING_SNAKE_CASE			  =parser.parse_args([]  )
														self.assertEqual(_a				,    Namespace(foo=_a				,    baz=_a				,    opt=_a  )  )

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

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

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

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




						def 							__UpperCamelCase				(       self					:      Any  )							->     Any:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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

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

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

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




						def 							__UpperCamelCase				(       self					:      Tuple  )							->     Any:


										"""simple docstring"""





										@dataclass
										class 						A__							:
																UpperCAmelCase             =						"toto"
										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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

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

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

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




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


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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

										self.argparsersEqual(_a				,    _a  )

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

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




						def 							__UpperCamelCase				(       self					:      Optional[int]  )							->     str:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =argparse.ArgumentParser()
										expected.add_argument('''--foo'''				,    default=_a				,    type=_a  )
										expected.add_argument('''--bar'''				,    default=_a				,    type=_a				,    help='''help message'''  )
										expected.add_argument('''--baz'''				,    default=_a				,    type=_a  )
										expected.add_argument('''--ces'''				,    nargs='''+'''				,    default=[]				,    type=_a  )
										expected.add_argument('''--des'''				,    nargs='''+'''				,    default=[]				,    type=_a  )

										_SCREAMING_SNAKE_CASE			  =[OptionalExample]
										if is_python_no_less_than_3_10:
														dataclass_types.append(_a  )

										for dataclass_type in dataclass_types:
														_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

														self.argparsersEqual(_a				,    _a  )

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

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




						def 							__UpperCamelCase				(       self					:      Optional[int]  )							->     int:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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




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


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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




						def 							__UpperCamelCase				(       self					:      List[str]  )							->     Optional[int]:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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

										_SCREAMING_SNAKE_CASE			  =parser.parse_dict(_a  )[0]
										_SCREAMING_SNAKE_CASE			  =BasicExample(**_a  )
										self.assertEqual(_a				,    _a  )




						def 							__UpperCamelCase				(       self					:      Union[str, Any]  )							->     str:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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

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




						def 							__UpperCamelCase				(       self					:      Union[str, Any]  )							->     Tuple:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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

										_SCREAMING_SNAKE_CASE			  =BasicExample(**_a  )
										self.assertEqual(_a				,    _a  )




						def 							__UpperCamelCase				(       self					:      Dict  )							->     List[str]:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )

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




						def 							__UpperCamelCase				(       self					:      Dict  )							->     Dict:


										"""simple docstring"""





										_SCREAMING_SNAKE_CASE			  =HfArgumentParser(_a  )
										self.assertIsNotNone(_a  )



from typing import TYPE_CHECKING

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


snake_case_						:			List[Any]													=							{
    '''configuration_graphormer''': ['''GRAPHORMER_PRETRAINED_CONFIG_ARCHIVE_MAP''', '''GraphormerConfig'''],
}

try:
							if not is_torch_available():
														raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
							pass
else:
							snake_case_						:			str													=							[
							    '''GRAPHORMER_PRETRAINED_MODEL_ARCHIVE_LIST''',
							    '''GraphormerForGraphClassification''',
							    '''GraphormerModel''',
							    '''GraphormerPreTrainedModel''',
							]


if TYPE_CHECKING:
							from .configuration_graphormer import GRAPHORMER_PRETRAINED_CONFIG_ARCHIVE_MAP, GraphormerConfig

							try:
														if not is_torch_available():
																					raise OptionalDependencyNotAvailable()
							except OptionalDependencyNotAvailable:
														pass
							else:
														from .modeling_graphormer import (
														    GRAPHORMER_PRETRAINED_MODEL_ARCHIVE_LIST,
														    GraphormerForGraphClassification,
														    GraphormerModel,
														    GraphormerPreTrainedModel,
														)


else:
							import sys

							snake_case_						:			Union[str, Any]													=							_LazyModule(__name__, globals()['''__file__'''], _import_structure, module_spec=__spec__)