




"""simple docstring"""





import inspect

from diffusers.utils import is_flax_available
from diffusers.utils.testing_utils import require_flax


if is_flax_available():
							import jax

@require_flax
class    __lowercase							:


				'''simple docstring'''


				def    _lowerCamelCase       ( self :  List[Any]   ):
										snake_case__      ,						snake_case__							:       List[Any]				      =       self.prepare_init_args_and_inputs_for_common()

										snake_case__							:       Any				      =       self.model_class(**SCREAMING_SNAKE_CASE_   )
										snake_case__							:       Union[str, Any]				      =       model.init(inputs_dict['''prng_key''']   ,inputs_dict['''sample''']   )
										jax.lax.stop_gradient(SCREAMING_SNAKE_CASE_   )

										snake_case__							:       Union[str, Any]				      =       model.apply(SCREAMING_SNAKE_CASE_   ,inputs_dict['''sample''']   )

										if isinstance(SCREAMING_SNAKE_CASE_   ,SCREAMING_SNAKE_CASE_   ):
																snake_case__							:       Dict				      =       output.sample

										self.assertIsNotNone(SCREAMING_SNAKE_CASE_   )
										snake_case__							:       Dict				      =       inputs_dict['''sample'''].shape
										self.assertEqual(output.shape   ,SCREAMING_SNAKE_CASE_   ,'''Input and output shapes do not match'''   )

				def    _lowerCamelCase       ( self :  Dict   ):
										snake_case__      ,						snake_case__							:       int				      =       self.prepare_init_args_and_inputs_for_common()

										snake_case__							:       Any				      =       1_6
										snake_case__							:       int				      =       (1_6, 3_2)

										snake_case__							:       Optional[Any]				      =       self.model_class(**SCREAMING_SNAKE_CASE_   )
										snake_case__							:       Any				      =       model.init(inputs_dict['''prng_key''']   ,inputs_dict['''sample''']   )
										jax.lax.stop_gradient(SCREAMING_SNAKE_CASE_   )

										snake_case__							:       List[str]				      =       model.apply(SCREAMING_SNAKE_CASE_   ,inputs_dict['''sample''']   )

										if isinstance(SCREAMING_SNAKE_CASE_   ,SCREAMING_SNAKE_CASE_   ):
																snake_case__							:       Optional[int]				      =       output.sample

										self.assertIsNotNone(SCREAMING_SNAKE_CASE_   )
										snake_case__							:       Tuple				      =       inputs_dict['''sample'''].shape
										self.assertEqual(output.shape   ,SCREAMING_SNAKE_CASE_   ,'''Input and output shapes do not match'''   )





				def    _lowerCamelCase       ( self :  List[Any]   ):
										snake_case__							:       int				      =       '''kwargs''' in inspect.signature(self.model_class.__init__   ).parameters
										snake_case__							:       List[Any]				      =       len(self.model_class._deprecated_kwargs   ) > 0

										if has_kwarg_in_model_class and not has_deprecated_kwarg:
																raise ValueError(
																    F'''{self.model_class} has `**kwargs` in its __init__ method but has not defined any deprecated kwargs'''
																    ''' under the `_deprecated_kwargs` class attribute. Make sure to either remove `**kwargs` if there are'''
																    ''' no deprecated arguments or add the deprecated argument with `_deprecated_kwargs ='''
																    ''' [<deprecated_argument>]`'''   )

										if not has_kwarg_in_model_class and has_deprecated_kwarg:
																raise ValueError(
																    F'''{self.model_class} doesn\'t have `**kwargs` in its __init__ method but has defined deprecated kwargs'''
																    ''' under the `_deprecated_kwargs` class attribute. Make sure to either add the `**kwargs` argument to'''
																    F''' {self.model_class}.__init__ if there are deprecated arguments or remove the deprecated argument'''
																    ''' from `_deprecated_kwargs = [<deprecated_argument>]`'''   )








"""simple docstring"""





_lowercase						   =     (1, 1_53, 3_70, 3_71, 16_34, 24_67_80_51, 1_15_13_22_19_01_87_63_99_25_65_09_55_97_97_39_71_52_24_01)
_lowercase						   =     (-1_53, -1, 0, 1.2, 2_00, "A", [], {}, None)

def 		lowerCAmelCase				(			lowerCamelCase		):
						'''simple docstring'''


						if not isinstance(lowerCamelCase   ,      lowerCamelCase		) or n < 1:
												return False

						# Initialization of sum and number of digits.
						snake_case__							:       Optional[Any]				      =       0
						snake_case__							:       Union[str, Any]				      =       0
						snake_case__							:       List[str]				      =       n
						# Calculation of digits of the number
						while temp > 0:
												number_of_digits += 1
												temp //= 10
						# Dividing number into separate digits and find Armstrong number
						snake_case__							:       Optional[int]				      =       n
						while temp > 0:
												snake_case__							:       Optional[int]				      =       temp % 10
												total += rem**number_of_digits
												temp //= 10
						return n == total

def 		lowerCAmelCase				(			lowerCamelCase		):
						'''simple docstring'''


						if not isinstance(lowerCamelCase   ,      lowerCamelCase		) or n < 1:
												return False

						# Init a "histogram" of the digits
						snake_case__							:       Union[str, Any]				      =       [0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
						snake_case__							:       int				      =       0
						snake_case__							:       int				      =       0
						snake_case__							:       Dict				      =       n
						while temp > 0:
												snake_case__      ,						snake_case__							:       Optional[int]				      =       divmod(lowerCamelCase   ,      10		)
												digit_histogram[rem] += 1
												digit_total += 1

						for cnt, i in zip(lowerCamelCase   ,      range(len(lowerCamelCase		)		)		):
												total += cnt * i**digit_total

						return n == total

def 		lowerCAmelCase				(			lowerCamelCase		):
						'''simple docstring'''


						if not isinstance(lowerCamelCase   ,      lowerCamelCase		) or n < 1:
												return False
						snake_case__							:       Union[str, Any]				      =       len(str(lowerCamelCase		)		)  # the power that all digits will be raised to
						# check if sum of each digit multiplied expo times is equal to number
						return n == sum(int(lowerCamelCase		) ** expo for i in str(lowerCamelCase		)		)

def 		lowerCAmelCase				(			):
						'''simple docstring'''


						snake_case__							:       Dict				      =       int(input('''Enter an integer to see if it is an Armstrong number: '''		).strip()		)
						print(f'''{num} is {'' if armstrong_number(lowerCamelCase		) else 'not '}an Armstrong number.'''		)
						print(f'''{num} is {'' if narcissistic_number(lowerCamelCase		) else 'not '}an Armstrong number.'''		)
						print(f'''{num} is {'' if pluperfect_number(lowerCamelCase		) else 'not '}an Armstrong number.'''		)


if __name__ == "__main__":
							import doctest

							doctest.testmod()
							main()


