



"""simple docstring"""






from typing import List, Optional, Tuple, Union

from . import logging
from .import_utils import is_torch_available, is_torch_version


if is_torch_available():
    import torch

lowerCamelCase				:     Any								=  logging.get_logger(__name__)  # pylint: disable=invalid-name

try:
    from torch._dynamo import allow_in_graph as maybe_allow_in_graph
except (ImportError, ModuleNotFoundError):


    def 							A__ (     cls    ):


      '''simple docstring'''


      return cls


def 							A__ (     UpperCamelCase__       ,							UpperCamelCase__ = None       ,							UpperCamelCase__ = None       ,							UpperCamelCase__ = None       ,							UpperCamelCase__ = None       ,							):


  '''simple docstring'''


  _SCREAMING_SNAKE_CASE        =			device
  _SCREAMING_SNAKE_CASE        =			shape[0]

  _SCREAMING_SNAKE_CASE        =			layout or torch.strided
  _SCREAMING_SNAKE_CASE        =			device or torch.device('''cpu'''    )

  if generator is not None:
    _SCREAMING_SNAKE_CASE        =			generator.device.type if not isinstance(_snake_case       ,							_snake_case    ) else generator[0].device.type
    if gen_device_type != device.type and gen_device_type == "cpu":
      _SCREAMING_SNAKE_CASE        =			'''cpu'''
      if device != "mps":
        logger.info(
            F'''The passed generator was created on \'cpu\' even though a tensor on {device} was expected.'''
            F''' Tensors will be created on \'cpu\' and then moved to {device}. Note that one can probably'''
            F''' slighly speed up this function by passing a generator that was created on the {device} device.'''    )
    elif gen_device_type != device.type and gen_device_type == "cuda":
      raise ValueError(F'''Cannot generate a {device} tensor from a generator of type {gen_device_type}.'''    )

  if isinstance(_snake_case       ,							_snake_case    ):
    _SCREAMING_SNAKE_CASE        =			(1,) + shape[1:]
    _SCREAMING_SNAKE_CASE        =			[
        torch.randn(_snake_case       ,							generator=generator[i]       ,							device=_snake_case       ,							dtype=_snake_case       ,							layout=_snake_case    )
        for i in range(_snake_case    )
    ]
    _SCREAMING_SNAKE_CASE        =			torch.cat(_snake_case       ,							dim=0    ).to(_snake_case    )
  else:
    _SCREAMING_SNAKE_CASE        =			torch.randn(_snake_case       ,							generator=_snake_case       ,							device=_snake_case       ,							dtype=_snake_case       ,							layout=_snake_case    ).to(_snake_case    )

  return latents







def 							A__ (     UpperCamelCase__    ):


  '''simple docstring'''


  if is_torch_version('''<'''       ,							'''2.0.0'''    ) or not hasattr(_snake_case       ,							'''_dynamo'''    ):
    return False
  return isinstance(_snake_case       ,							torch._dynamo.eval_frame.OptimizedModule    )




"""simple docstring"""
import unittest

from diffusers import __version__
from diffusers.utils import deprecate



class      _UpperCAmelCase					(				unittest.TestCase							):






		'''simple docstring'''

		lowerCamelCase__           ='.'.join([str(int(__version__.split('.'							)[0]							) + 1							)] + __version__.split('.'							)[1:]							)
		lowerCamelCase__           ='0.0.1'



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		int         =							{'''deprecated_arg''': 4}

			with self.assertWarns(a_   ) as warning:
				__snake_case      :		Union[str, Any]         =							deprecate('''deprecated_arg'''							,			self.higher_version							,			'''message'''							,			take_from=a_   )

			assert output == 4
			assert (
			    str(warning.warning   )
			    == f"""The `deprecated_arg` argument is deprecated and will be removed in version {self.higher_version}."""
			    " message"
			)



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		Tuple         =							{'''deprecated_arg''': 4}

			with self.assertWarns(a_   ) as warning:
				__snake_case      :		Dict         =							deprecate(('''deprecated_arg''', self.higher_version, '''message''')							,			take_from=a_   )

			assert output == 4
			assert (
			    str(warning.warning   )
			    == f"""The `deprecated_arg` argument is deprecated and will be removed in version {self.higher_version}."""
			    " message"
			)



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		Union[str, Any]         =							{'''deprecated_arg_1''': 4, '''deprecated_arg_2''': 8}
			with self.assertWarns(a_   ) as warning:
				__snake_case      ,				__snake_case      :		List[Any]         =							deprecate(
				    ('''deprecated_arg_1''', self.higher_version, '''Hey''')							,			('''deprecated_arg_2''', self.higher_version, '''Hey''')							,			take_from=a_							,			)
			assert output_a == 4
			assert output_a == 8
			assert (
			    str(warning.warnings[0].message   )
			    == "The `deprecated_arg_1` argument is deprecated and will be removed in version"
			    f""" {self.higher_version}. Hey"""
			)
			assert (
			    str(warning.warnings[1].message   )
			    == "The `deprecated_arg_2` argument is deprecated and will be removed in version"
			    f""" {self.higher_version}. Hey"""
			)



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		int         =							{'''deprecated_arg''': 4}

			with self.assertRaises(a_   ) as error:
				deprecate(('''wrong_arg''', self.higher_version, '''message''')							,			take_from=a_   )

			assert "test_deprecate_function_incorrect_arg in" in str(error.exception   )
			assert "line" in str(error.exception   )
			assert "got an unexpected keyword argument `deprecated_arg`" in str(error.exception   )



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			with self.assertWarns(a_   ) as warning:
				deprecate(('''deprecated_arg''', self.higher_version, '''message''')   )

			assert (
			    str(warning.warning   )
			    == f"""`deprecated_arg` is deprecated and will be removed in version {self.higher_version}. message"""
			)



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			with self.assertWarns(a_   ) as warning:
				deprecate(
				    ('''deprecated_arg_1''', self.higher_version, '''Hey''')							,			('''deprecated_arg_2''', self.higher_version, '''Hey''')							,			)
			assert (
			    str(warning.warnings[0].message   )
			    == f"""`deprecated_arg_1` is deprecated and will be removed in version {self.higher_version}. Hey"""
			)
			assert (
			    str(warning.warnings[1].message   )
			    == f"""`deprecated_arg_2` is deprecated and will be removed in version {self.higher_version}. Hey"""
			)



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			class      _UpperCAmelCase					:






					'''simple docstring'''

					lowerCamelCase__           =5

			with self.assertWarns(a_   ) as warning:
				__snake_case      :		List[Any]         =							deprecate(('''arg''', self.higher_version, '''message''')							,			take_from=Args()   )

			assert arg == 5
			assert (
			    str(warning.warning   )
			    == f"""The `arg` attribute is deprecated and will be removed in version {self.higher_version}. message"""
			)



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			class      _UpperCAmelCase					:






					'''simple docstring'''

					lowerCamelCase__           =5
					lowerCamelCase__           =7

			with self.assertWarns(a_   ) as warning:
				__snake_case      ,				__snake_case      :		List[str]         =							deprecate(
				    ('''arg''', self.higher_version, '''message''')							,			('''foo''', self.higher_version, '''message''')							,			('''does not exist''', self.higher_version, '''message''')							,			take_from=Args()							,			)

			assert arg_a == 5
			assert arg_a == 7
			assert (
			    str(warning.warning   )
			    == f"""The `arg` attribute is deprecated and will be removed in version {self.higher_version}. message"""
			)
			assert (
			    str(warning.warnings[0].message   )
			    == f"""The `arg` attribute is deprecated and will be removed in version {self.higher_version}. message"""
			)
			assert (
			    str(warning.warnings[1].message   )
			    == f"""The `foo` attribute is deprecated and will be removed in version {self.higher_version}. message"""
			)



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			__snake_case      :		str         =							{'''deprecated_arg''': 4}

			with self.assertRaises(a_   ) as error:
				deprecate(('''wrong_arg''', self.lower_version, '''message''')							,			take_from=a_   )

			assert (
			    str(error.exception   )
			    == "The deprecation tuple ('wrong_arg', '0.0.1', 'message') should be removed since diffusers' version"
			    f""" {__version__} is >= {self.lower_version}"""
			)



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			with self.assertWarns(a_   ) as warning:
				deprecate(('''deprecated_arg''', self.higher_version, '''This message is better!!!''')							,			standard_warn=a_   )

			assert str(warning.warning   ) == "This message is better!!!"



		def 							SCREAMING_SNAKE_CASE    (self   ):





			'''simple docstring'''

			with self.assertWarns(a_   ) as warning:
				deprecate(('''deprecated_arg''', self.higher_version, '''This message is better!!!''')							,			standard_warn=a_   )
			assert str(warning.warning   ) == "This message is better!!!"
			assert "diffusers/tests/others/test_utils.py" in warning.filename





