



'''simple docstring'''






import os

from packaging import version

from .. import __version__
from .accelerate_utils import apply_forward_hook
from .constants import (
    CONFIG_NAME,
    DEPRECATED_REVISION_ARGS,
    DIFFUSERS_CACHE,
    DIFFUSERS_DYNAMIC_MODULE_NAME,
    FLAX_WEIGHTS_NAME,
    HF_MODULES_CACHE,
    HUGGINGFACE_CO_RESOLVE_ENDPOINT,
    ONNX_EXTERNAL_WEIGHTS_NAME,
    ONNX_WEIGHTS_NAME,
    SAFETENSORS_WEIGHTS_NAME,
    TEXT_ENCODER_ATTN_MODULE,
    WEIGHTS_NAME,
)
from .deprecation_utils import deprecate
from .doc_utils import replace_example_docstring
from .dynamic_modules_utils import get_class_from_dynamic_module
from .hub_utils import (
    HF_HUB_OFFLINE,
    _add_variant,
    _get_model_file,
    extract_commit_hash,
    http_user_agent,
)
from .import_utils import (
    BACKENDS_MAPPING,
    ENV_VARS_TRUE_AND_AUTO_VALUES,
    ENV_VARS_TRUE_VALUES,
    USE_JAX,
    USE_TF,
    USE_TORCH,
    DummyObject,
    OptionalDependencyNotAvailable,
    is_accelerate_available,
    is_accelerate_version,
    is_bsa_available,
    is_flax_available,
    is_ftfy_available,
    is_inflect_available,
    is_invisible_watermark_available,
    is_k_diffusion_available,
    is_k_diffusion_version,
    is_librosa_available,
    is_note_seq_available,
    is_omegaconf_available,
    is_onnx_available,
    is_safetensors_available,
    is_scipy_available,
    is_tensorboard_available,
    is_tf_available,
    is_torch_available,
    is_torch_version,
    is_torchsde_available,
    is_transformers_available,
    is_transformers_version,
    is_unidecode_available,
    is_wandb_available,
    is_xformers_available,
    requires_backends,
)
from .logging import get_logger
from .outputs import BaseOutput
from .pil_utils import PIL_INTERPOLATION, numpy_to_pil, pt_to_pil
from .torch_utils import is_compiled_module, randn_tensor


if is_torch_available():
   from .testing_utils import (
       floats_tensor,
       load_hf_numpy,
       load_image,
       load_numpy,
       load_pt,
       nightly,
       parse_flag_from_env,
       print_tensor_test,
       require_torch_a,
       require_torch_gpu,
       skip_mps,
       slow,
       torch_all_close,
       torch_device,
   )
   from .torch_utils import maybe_allow_in_graph

from .testing_utils import export_to_gif, export_to_video


A_				:					Any							=      get_logger(__name__)




def        snake_case_   (	lowerCAmelCase_						)->						List[str]:





       '''simple docstring'''


       if version.parse(lowerCAmelCase_						) < version.parse(lowerCAmelCase_						):
              if "dev" in min_version:
                     _UpperCAmelCase    :		Any     	=					(
                         """This example requires a source install from HuggingFace diffusers (see """
                         """`https://huggingface.co/docs/diffusers/installation#install-from-source`),"""
                     )
              else:
                     _UpperCAmelCase    :		Dict     	=					F'''This example requires a minimum version of {min_version},'''
              error_message += F''' but the version found is {__version__}.\n'''
              raise ImportError(lowerCAmelCase_						)







"""simple docstring"""
import unittest
from dataclasses import dataclass
from typing import Optional

from transformers.utils import ModelOutput



@dataclass
class      _UpperCAmelCase					(				__snake_case							):






		'''simple docstring'''

		lowerCamelCase__           =42
		lowerCamelCase__           =None
		lowerCamelCase__           =None



class      _UpperCAmelCase					(				unittest.TestCase							):






		'''simple docstring'''

		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		Union[str, Any]         =							ModelOutputTest(a=30   )
			self.assertEqual(x.a							,			30   )
			self.assertIsNone(x.b   )
			self.assertIsNone(x.c   )
			with self.assertRaises(a_   ):
				__snake_case      :		str         =							x.d



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		Union[str, Any]         =							ModelOutputTest(a=30							,			b=10   )
			self.assertEqual(x[0]							,			30   )
			self.assertEqual(x[1]							,			10   )
			self.assertEqual(x[:2]							,			(30, 10)   )
			self.assertEqual(x[:]							,			(30, 10)   )

			__snake_case      :		List[Any]         =							ModelOutputTest(a=30							,			c=10   )
			self.assertEqual(x[0]							,			30   )
			self.assertEqual(x[1]							,			10   )
			self.assertEqual(x[:2]							,			(30, 10)   )
			self.assertEqual(x[:]							,			(30, 10)   )



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		Optional[int]         =							ModelOutputTest(a=30							,			b=10   )
			self.assertEqual(x['''a''']							,			30   )
			self.assertEqual(x['''b''']							,			10   )
			with self.assertRaises(a_   ):
				__snake_case      :		Optional[int]         =							x['''c''']

			__snake_case      :		Any         =							ModelOutputTest(a=30							,			c=10   )
			self.assertEqual(x['''a''']							,			30   )
			self.assertEqual(x['''c''']							,			10   )
			with self.assertRaises(a_   ):
				__snake_case      :		Optional[int]         =							x['''b''']



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		Optional[int]         =							ModelOutputTest(a=30   )
			self.assertEqual(list(x.keys()   )							,			['''a''']   )
			self.assertEqual(list(x.values()   )							,			[30]   )
			self.assertEqual(list(x.items()   )							,			[('''a''', 30)]   )
			self.assertEqual(list(a_   )							,			['''a''']   )

			__snake_case      :		Optional[int]         =							ModelOutputTest(a=30							,			b=10   )
			self.assertEqual(list(x.keys()   )							,			['''a''', '''b''']   )
			self.assertEqual(list(x.values()   )							,			[30, 10]   )
			self.assertEqual(list(x.items()   )							,			[('''a''', 30), ('''b''', 10)]   )
			self.assertEqual(list(a_   )							,			['''a''', '''b''']   )

			__snake_case      :		Optional[Any]         =							ModelOutputTest(a=30							,			c=10   )
			self.assertEqual(list(x.keys()   )							,			['''a''', '''c''']   )
			self.assertEqual(list(x.values()   )							,			[30, 10]   )
			self.assertEqual(list(x.items()   )							,			[('''a''', 30), ('''c''', 10)]   )
			self.assertEqual(list(a_   )							,			['''a''', '''c''']   )

			with self.assertRaises(a_   ):
				__snake_case      :		Optional[int]         =							x.update({'''d''': 20}   )
			with self.assertRaises(a_   ):
				del x["a"]
			with self.assertRaises(a_   ):
				__snake_case      :		Union[str, Any]         =							x.pop('''a'''   )
			with self.assertRaises(a_   ):
				__snake_case      :		Optional[Any]         =							x.setdefault('''d'''							,			32   )



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		List[str]         =							ModelOutputTest(a=30   )
			__snake_case      :		Tuple         =							10
			self.assertEqual(x.a							,			10   )
			self.assertEqual(x['''a''']							,			10   )



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		List[str]         =							ModelOutputTest(a=30   )
			__snake_case      :		Any         =							10
			self.assertEqual(x.a							,			10   )
			self.assertEqual(x['''a''']							,			10   )



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		int         =							ModelOutputTest({'''a''': 30, '''b''': 10}   )
			self.assertEqual(list(x.keys()   )							,			['''a''', '''b''']   )
			self.assertEqual(x.a							,			30   )
			self.assertEqual(x.b							,			10   )



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		Any         =							ModelOutputTest([('''a''', 30), ('''b''', 10)]   )
			self.assertEqual(list(x.keys()   )							,			['''a''', '''b''']   )
			self.assertEqual(x.a							,			30   )
			self.assertEqual(x.b							,			10   )

			with self.assertRaises(a_   ):
				__snake_case      :		List[Any]         =							ModelOutputTest([('''a''', 30), (10, 10)]   )

			__snake_case      :		Union[str, Any]         =							ModelOutputTest(a=(30, 30)   )
			self.assertEqual(list(x.keys()   )							,			['''a''']   )
			self.assertEqual(x.a							,			(30, 30)   )





