'''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 				_lowercase	(      lowerCAmelCase		,unittest.TestCase   ):
      '''simple docstring'''


      UpperCAmelCase_      :      Optional[int]      							=   FlaxAutoencoderKL




      @property
      def 		lowerCAmelCase__					(				self							)			->			Dict:



         '''simple docstring'''



         UpperCAmelCase__							:				List[Any]   				=  4
         UpperCAmelCase__							:				List[Any]   				=  3
         UpperCAmelCase__							:				int   				=  (32, 32)

         UpperCAmelCase__							:				List[Any]   				=  jax.random.PRNGKey(0							)
         UpperCAmelCase__							:				Any   				=  jax.random.uniform(lowerCamelCase_					,((batch_size, num_channels) + sizes)							)

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




      def 		lowerCAmelCase__					(				self							)			->			Dict:



         '''simple docstring'''



         UpperCAmelCase__							:				str   				=  {
             '''block_out_channels''': [32, 64],
             '''in_channels''': 3,
             '''out_channels''': 3,
             '''down_block_types''': ['''DownEncoderBlock2D''', '''DownEncoderBlock2D'''],
             '''up_block_types''': ['''UpDecoderBlock2D''', '''UpDecoderBlock2D'''],
             '''latent_channels''': 4,
         }
         UpperCAmelCase__							:				Optional[Any]   				=  self.dummy_input
         return init_dict, inputs_dict





'''simple docstring'''

def __UpperCamelCase(   _A					:				str  ,			_A					:				str    ):
   '''simple docstring'''
   UpperCAmelCase__							:				int   				=  len(_A    )
   UpperCAmelCase__							:				int   				=  len(_A    )
   UpperCAmelCase__							:				int   				=  (
       first_str_length if first_str_length > second_str_length else second_str_length
   )
   UpperCAmelCase__							:				list   				=  []
   for char_count in range(_A    ):
      if char_count < first_str_length:
         output_list.append(first_str[char_count]    )
      if char_count < second_str_length:
         output_list.append(second_str[char_count]    )
   return "".join(_A    )


if __name__ == "__main__":
  print(alternative_string_arrange('AB', 'XYZ'), end=' ')




