
"""simple docstring"""

import unittest

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

from .test_modeling_common_flax import FlaxModelTesterMixin


if is_flax_available():
   import jax
@require_flax
class   _UpperCAmelCase				(		SCREAMING_SNAKE_CASE_  ,      unittest.TestCase				):
      __SCREAMING_SNAKE_CASE :				str				   =						FlaxAutoencoderKL




      @property
      def      a_   (				self				)    ->       int:
          UpperCAmelCase          =       4
          UpperCAmelCase          =       3
          UpperCAmelCase          =       (3_2, 3_2)

          UpperCAmelCase          =       jax.random.PRNGKey(0				)
          UpperCAmelCase          =       jax.random.uniform(lowercase_ ,							((batch_size, num_channels) + sizes)				)

          return {"sample": image, "prng_key": prng_key}




      def      a_   (				self				)    ->       Dict:
          UpperCAmelCase          =       {
              'block_out_channels': [3_2, 6_4],
              'in_channels': 3,
              'out_channels': 3,
              'down_block_types': ['DownEncoderBlock2D', 'DownEncoderBlock2D'],
              'up_block_types': ['UpDecoderBlock2D', 'UpDecoderBlock2D'],
              'latent_channels': 4,
          }
          UpperCAmelCase          =       self.dummy_input
          return init_dict, inputs_dict





"""simple docstring"""

import math




def      lowercase__    (       lowerCAmelCase					:       str			,					lowerCAmelCase					:       Optional[Any]			)			->   List[Any]:






    """simple docstring"""




    if 0 not in (x, y):
        # We use the relation x^y = y*log10(x), where 10 is the base.
        return y * math.logaa(lowerCAmelCase			)
    else:
        if x == 0:  # 0 raised to any number is 0
            return 0
        elif y == 0:
            return 1  # any number raised to 0 is 1
    raise AssertionError('This should never happen'			)


if __name__ == "__main__":  # Main function
   # Read two numbers from input and typecast them to int using map function.
   # Here x is the base and y is the power.
   SCREAMING_SNAKE_CASE_            =    '''Enter the base and the power separated by a comma: '''
   SCREAMING_SNAKE_CASE_       ,    SCREAMING_SNAKE_CASE_            =    map(int, input(prompt).split(''','''))
   SCREAMING_SNAKE_CASE_       ,    SCREAMING_SNAKE_CASE_            =    map(int, input(prompt).split(''','''))

   # We find the log of each number, using the function res(), which takes two
   # arguments.
   SCREAMING_SNAKE_CASE_            =    res(xa, ya)
   SCREAMING_SNAKE_CASE_            =    res(xa, ya)

   # We check for the largest number
   if resa > resa:
      print('''Largest number is''', xa, '''^''', ya)
   elif resa > resa:
      print('''Largest number is''', xa, '''^''', ya)
   else:
      print('''Both are equal''')



