



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      __lowerCAmelCase       :



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



						"""simple docstring"""






						_UpperCAmelCase			,	_UpperCAmelCase									=  self.prepare_init_args_and_inputs_for_common()

						_UpperCAmelCase									=  self.model_class(**A)
						_UpperCAmelCase									=  model.init(inputs_dict['prng_key']					,						inputs_dict['sample'])
						jax.lax.stop_gradient(A)

						_UpperCAmelCase									=  model.apply(A					,						inputs_dict['sample'])

						if isinstance(A					,						A):
									_UpperCAmelCase									=  output.sample

						self.assertIsNotNone(A)
						_UpperCAmelCase									=  inputs_dict['sample'].shape
						self.assertEqual(output.shape					,						A					,						'Input and output shapes do not match')



			def        _lowerCamelCase			(		self  :    int)  ->       Any:



						"""simple docstring"""






						_UpperCAmelCase			,	_UpperCAmelCase									=  self.prepare_init_args_and_inputs_for_common()

						_UpperCAmelCase									=  16
						_UpperCAmelCase									=  (16, 32)

						_UpperCAmelCase									=  self.model_class(**A)
						_UpperCAmelCase									=  model.init(inputs_dict['prng_key']					,						inputs_dict['sample'])
						jax.lax.stop_gradient(A)

						_UpperCAmelCase									=  model.apply(A					,						inputs_dict['sample'])

						if isinstance(A					,						A):
									_UpperCAmelCase									=  output.sample

						self.assertIsNotNone(A)
						_UpperCAmelCase									=  inputs_dict['sample'].shape
						self.assertEqual(output.shape					,						A					,						'Input and output shapes do not match')



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



						"""simple docstring"""






						_UpperCAmelCase									=  'kwargs' in inspect.signature(self.model_class.__init__).parameters
						_UpperCAmelCase									=  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>]`')





UpperCAmelCase__  =			10**8





def 				A      (	_UpperCAmelCase							:  int       )							->					bool:







			'''simple docstring'''
			if n % 10 == 0:
						return False
			_UpperCAmelCase									=  str(_UpperCAmelCase       )
			return s == s[::-1]





def 				A      (	)							->					int:







			'''simple docstring'''
			_UpperCAmelCase									=  set()
			_UpperCAmelCase									=  1
			_UpperCAmelCase									=  5
			while sum_squares < LIMIT:
						_UpperCAmelCase									=  first_square + 1
						while sum_squares < LIMIT:
									if is_palindrome(_UpperCAmelCase       ):
												answer.add(_UpperCAmelCase       )
									last_square += 1
									sum_squares += last_square**2
						first_square += 1
						_UpperCAmelCase									=  first_square**2 + (first_square + 1) ** 2

			return sum(_UpperCAmelCase       )


if __name__ == "__main__":
							print(solution())
