


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
a_							=    sys.version_info >= (3, 10)


def        _a  (     UpperCamelCase_					:				Union[str, Any]=None    ,   UpperCamelCase_					:				str=None   ) ->   str:
						"""simple docstring"""







						return field(default_factory=lambda: default    ,   metadata=UpperCamelCase_   )


@dataclass
class      lowercase__     :
	a_         =42
	a_         =42
	a_         =42
	a_         =42





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





@dataclass
class      lowercase__     :
	a_         =False
	a_         =True
	a_         =None





class      lowercase__     (			_a	):
	a_         ="""titi"""
	a_         ="""toto"""





class      lowercase__     (			_a	):
	a_         ="""titi"""
	a_         ="""toto"""
	a_         =42





@dataclass
class      lowercase__     :
	a_         ="toto"

	def 		UpperCAmelCase  (							self       )->					Tuple:







							'''simple docstring'''





							lowerCAmelCase__       =						BasicEnum(self.foo       )





@dataclass
class      lowercase__     :
	a_         ="toto"

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







							'''simple docstring'''





							lowerCAmelCase__       =						MixedTypeEnum(self.foo       )





@dataclass
class      lowercase__     :
	a_         =None
	a_         =field(default=_a,					metadata={"""help""": """help message"""}	)
	a_         =None
	a_         =list_field(default=[]	)
	a_         =list_field(default=[]	)





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





@dataclass
class      lowercase__     :
	a_         =field()
	a_         =field()
	a_         =field()

	def 		UpperCAmelCase  (							self       )->					Dict:







							'''simple docstring'''





							lowerCAmelCase__       =						BasicEnum(self.required_enum       )





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


if is_python_no_less_than_3_10:


	@dataclass
	class      lowercase__     :
		a_         =False
		a_         =True
		a_         =None





	@dataclass
	class      lowercase__     :
		a_         =None
		a_         =field(default=_a,					metadata={"""help""": """help message"""}	)
		a_         =None
		a_         =list_field(default=[]	)
		a_         =list_field(default=[]	)





class      lowercase__     (			unittest.TestCase	):

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







							'''simple docstring'''





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

													self.assertEqual(snake_case_		,    snake_case_       )

	def 		UpperCAmelCase  (							self       )->					Dict:







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

							lowerCAmelCase__       =						argparse.ArgumentParser()
							expected.add_argument("--foo"		,    type=snake_case_		,    required=snake_case_       )
							expected.add_argument("--bar"		,    type=snake_case_		,    required=snake_case_       )
							expected.add_argument("--baz"		,    type=snake_case_		,    required=snake_case_       )
							expected.add_argument("--flag"		,    type=snake_case_		,    default=snake_case_		,    const=snake_case_		,    nargs="?"       )
							self.argparsersEqual(snake_case_		,    snake_case_       )

							lowerCAmelCase__       =						["--foo", "1", "--baz", "quux", "--bar", "0.5"]
							((lowerCAmelCase__						)      ,													)       =						parser.parse_args_into_dataclasses(snake_case_		,    look_for_args_file=snake_case_       )
							self.assertFalse(example.flag       )

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







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

							lowerCAmelCase__       =						argparse.ArgumentParser()
							expected.add_argument("--foo"		,    default=42		,    type=snake_case_       )
							expected.add_argument("--baz"		,    default="toto"		,    type=snake_case_		,    help="help message"       )
							self.argparsersEqual(snake_case_		,    snake_case_       )

	def 		UpperCAmelCase  (							self       )->					Any:







							'''simple docstring'''





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

							lowerCAmelCase__       =						[WithDefaultBoolExample]
							if is_python_no_less_than_3_10:
													dataclass_types.append(snake_case_       )

							for dataclass_type in dataclass_types:
													lowerCAmelCase__       =						HfArgumentParser(snake_case_       )
													self.argparsersEqual(snake_case_		,    snake_case_       )

													lowerCAmelCase__       =						parser.parse_args([]       )
													self.assertEqual(snake_case_		,    Namespace(foo=snake_case_		,    baz=snake_case_		,    opt=snake_case_       )       )

													lowerCAmelCase__       =						parser.parse_args(["--foo", "--no_baz"]       )
													self.assertEqual(snake_case_		,    Namespace(foo=snake_case_		,    baz=snake_case_		,    opt=snake_case_       )       )

													lowerCAmelCase__       =						parser.parse_args(["--foo", "--baz"]       )
													self.assertEqual(snake_case_		,    Namespace(foo=snake_case_		,    baz=snake_case_		,    opt=snake_case_       )       )

													lowerCAmelCase__       =						parser.parse_args(["--foo", "True", "--baz", "True", "--opt", "True"]       )
													self.assertEqual(snake_case_		,    Namespace(foo=snake_case_		,    baz=snake_case_		,    opt=snake_case_       )       )

													lowerCAmelCase__       =						parser.parse_args(["--foo", "False", "--baz", "False", "--opt", "False"]       )
													self.assertEqual(snake_case_		,    Namespace(foo=snake_case_		,    baz=snake_case_		,    opt=snake_case_       )       )

	def 		UpperCAmelCase  (							self       )->					Any:







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

							lowerCAmelCase__       =						argparse.ArgumentParser()
							expected.add_argument(
							    "--foo"		,    default="toto"		,    choices=["titi", "toto", 42]		,    type=make_choice_type_function(["titi", "toto", 42]       )		,    )
							self.argparsersEqual(snake_case_		,    snake_case_       )

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

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

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

	def 		UpperCAmelCase  (							self       )->					Any:







							'''simple docstring'''





							@dataclass
							class      lowercase__     :
								a_         ="toto"





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

							lowerCAmelCase__       =						argparse.ArgumentParser()
							expected.add_argument(
							    "--foo"		,    default="toto"		,    choices=("titi", "toto", 42)		,    type=make_choice_type_function(["titi", "toto", 42]       )		,    )
							self.argparsersEqual(snake_case_		,    snake_case_       )

							lowerCAmelCase__       =						parser.parse_args([]       )
							self.assertEqual(args.foo		,    "toto"       )

							lowerCAmelCase__       =						parser.parse_args(["--foo", "titi"]       )
							self.assertEqual(args.foo		,    "titi"       )

							lowerCAmelCase__       =						parser.parse_args(["--foo", "42"]       )
							self.assertEqual(args.foo		,    42       )

	def 		UpperCAmelCase  (							self       )->					Tuple:







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

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

							self.argparsersEqual(snake_case_		,    snake_case_       )

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

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

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







							'''simple docstring'''





							lowerCAmelCase__       =						argparse.ArgumentParser()
							expected.add_argument("--foo"		,    default=snake_case_		,    type=snake_case_       )
							expected.add_argument("--bar"		,    default=snake_case_		,    type=snake_case_		,    help="help message"       )
							expected.add_argument("--baz"		,    default=snake_case_		,    type=snake_case_       )
							expected.add_argument("--ces"		,    nargs="+"		,    default=[]		,    type=snake_case_       )
							expected.add_argument("--des"		,    nargs="+"		,    default=[]		,    type=snake_case_       )

							lowerCAmelCase__       =						[OptionalExample]
							if is_python_no_less_than_3_10:
													dataclass_types.append(snake_case_       )

							for dataclass_type in dataclass_types:
													lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

													self.argparsersEqual(snake_case_		,    snake_case_       )

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

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

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







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

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

	def 		UpperCAmelCase  (							self       )->					int:







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

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

	def 		UpperCAmelCase  (							self       )->					str:







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

							lowerCAmelCase__       =						{
							    "foo": 12,
							    "bar": 3.14,
							    "baz": "42",
							    "flag": True,
							}

							lowerCAmelCase__       =						parser.parse_dict(snake_case_       )[0]
							lowerCAmelCase__       =						BasicExample(**snake_case_       )
							self.assertEqual(snake_case_		,    snake_case_       )

	def 		UpperCAmelCase  (							self       )->					Dict:







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

							lowerCAmelCase__       =						{
							    "foo": 12,
							    "bar": 3.14,
							    "baz": "42",
							    "flag": True,
							    "extra": 42,
							}

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

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







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

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

							lowerCAmelCase__       =						BasicExample(**snake_case_       )
							self.assertEqual(snake_case_		,    snake_case_       )

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







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )

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


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







							'''simple docstring'''





							lowerCAmelCase__       =						HfArgumentParser(snake_case_       )
							self.assertIsNotNone(snake_case_       )














# limitations under the License.

# NOTE: This file is deprecated and will be removed in a future version.
# It only exists so that temporarely `from diffusers.pipelines import DiffusionPipeline` works

from .pipelines import DiffusionPipeline, ImagePipelineOutput  # noqa: F401
from .utils import deprecate


deprecate(
    '''pipelines_utils''',
    '''0.22.0''',
    '''Importing `DiffusionPipeline` or `ImagePipelineOutput` from diffusers.pipeline_utils is deprecated. Please import from diffusers.pipelines.pipeline_utils instead.''',
    standard_warn=False,
    stacklevel=3,
)






