import tempfile
import unittest

from diffusers import (
    DDIMScheduler,
    DDPMScheduler,
    DPMSolverMultistepScheduler,
    EulerAncestralDiscreteScheduler,
    EulerDiscreteScheduler,
    PNDMScheduler,
    logging,
)
from diffusers.configuration_utils import ConfigMixin, register_to_config
from diffusers.utils.testing_utils import CaptureLogger






class     __UpperCAmelCase(    A__				):






 """simple docstring"""




 __magic_name__          =   """config.json"""
 @register_to_config
 def __init__(    self ,       __magic_name__=2 ,       __magic_name__=5 ,       __magic_name__=(2, 5) ,       __magic_name__="for diffusion" ,       __magic_name__=[1, 3] ,       ):

        """simple docstring"""

        pass






class     __UpperCAmelCase(    A__				):






 """simple docstring"""




 __magic_name__          =   """config.json"""
 @register_to_config
 def __init__(    self ,       __magic_name__=2 ,       __magic_name__=5 ,       __magic_name__=(2, 5) ,       __magic_name__="for diffusion" ,       __magic_name__=[1, 3] ,       ):

        """simple docstring"""

        pass






class     __UpperCAmelCase(    A__				):






 """simple docstring"""




 __magic_name__          =   """config.json"""
 @register_to_config
 def __init__(    self ,       __magic_name__=2 ,       __magic_name__=5 ,       __magic_name__=(2, 5) ,       __magic_name__="for diffusion" ,       __magic_name__=[1, 3] ,       __magic_name__=[1, 3] ,       ):

        """simple docstring"""

        pass






class     __UpperCAmelCase(    A__				):






 """simple docstring"""




 __magic_name__          =   """config.json"""
 @register_to_config
 def __init__(    self ,       __magic_name__=2 ,       __magic_name__=5 ,       __magic_name__=(2, 5) ,       __magic_name__="for diffusion" ,       __magic_name__=[1, 5] ,       __magic_name__=[5, 4] ,       ):

        """simple docstring"""

        pass






class     __UpperCAmelCase(    unittest.TestCase				):






 """simple docstring"""




 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        with self.assertRaises(lowerCAmelCase_      ):
               ConfigMixin.load_config('''dummy_path'''      )
 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        A_							:       str	      =				SampleObject()
        A_							:       Any	      =				obj.config
        assert config["a"] == 2
        assert config["b"] == 5
        assert config["c"] == (2, 5)
        assert config["d"] == "for diffusion"
        assert config["e"] == [1, 3]

        # init ignore private arguments
        A_							:       Dict	      =				SampleObject(_name_or_path='''lalala'''      )
        A_							:       Tuple	      =				obj.config
        assert config["a"] == 2
        assert config["b"] == 5
        assert config["c"] == (2, 5)
        assert config["d"] == "for diffusion"
        assert config["e"] == [1, 3]

        # can override default
        A_							:       Any	      =				SampleObject(c=6      )
        A_							:       str	      =				obj.config
        assert config["a"] == 2
        assert config["b"] == 5
        assert config["c"] == 6
        assert config["d"] == "for diffusion"
        assert config["e"] == [1, 3]

        # can use positional arguments.
        A_							:       int	      =				SampleObject(1 ,       c=6      )
        A_							:       int	      =				obj.config
        assert config["a"] == 1
        assert config["b"] == 5
        assert config["c"] == 6
        assert config["d"] == "for diffusion"
        assert config["e"] == [1, 3]
 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        A_							:       Dict	      =				SampleObject()
        A_							:       Optional[int]	      =				obj.config

        assert config["a"] == 2
        assert config["b"] == 5
        assert config["c"] == (2, 5)
        assert config["d"] == "for diffusion"
        assert config["e"] == [1, 3]

        with tempfile.TemporaryDirectory() as tmpdirname:
               obj.save_config(lowerCAmelCase_      )
               A_							:       List[Any]	      =				SampleObject.from_config(SampleObject.load_config(lowerCAmelCase_      )      )
               A_							:       Any	      =				new_obj.config

        # unfreeze configs
        A_							:       str	      =				dict(lowerCAmelCase_      )
        A_							:       List[str]	      =				dict(lowerCAmelCase_      )

        assert config.pop('''c'''      ) == (2, 5)  # instantiated as tuple
        assert new_config.pop('''c'''      ) == [2, 5]  # saved & loaded as list because of json
        config.pop('''_use_default_values'''      )
        assert config == new_config
 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        A_							:       Union[str, Any]	      =				logging.get_logger('''diffusers.configuration_utils'''      )
        # 30 for warning
        logger.setLevel(30      )

        with CaptureLogger(lowerCAmelCase_      ) as cap_logger:
               A_							:       Any	      =				DDIMScheduler.from_pretrained(
                   '''hf-internal-testing/tiny-stable-diffusion-torch''' ,       subfolder='''scheduler'''      )

        assert ddim.__class__ == DDIMScheduler
        # no warning should be thrown
        assert cap_logger.out == ""
 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        A_							:       Tuple	      =				logging.get_logger('''diffusers.configuration_utils'''      )
        # 30 for warning
        logger.setLevel(30      )

        with CaptureLogger(lowerCAmelCase_      ) as cap_logger:
               A_							:       List[str]	      =				EulerDiscreteScheduler.from_pretrained(
                   '''hf-internal-testing/tiny-stable-diffusion-torch''' ,       subfolder='''scheduler'''      )

        assert euler.__class__ == EulerDiscreteScheduler
        # no warning should be thrown
        assert cap_logger.out == ""
 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        A_							:       Dict	      =				logging.get_logger('''diffusers.configuration_utils'''      )
        # 30 for warning
        logger.setLevel(30      )

        with CaptureLogger(lowerCAmelCase_      ) as cap_logger:
               A_							:       Optional[Any]	      =				EulerAncestralDiscreteScheduler.from_pretrained(
                   '''hf-internal-testing/tiny-stable-diffusion-torch''' ,       subfolder='''scheduler'''      )

        assert euler.__class__ == EulerAncestralDiscreteScheduler
        # no warning should be thrown
        assert cap_logger.out == ""
 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        A_							:       Any	      =				logging.get_logger('''diffusers.configuration_utils'''      )
        # 30 for warning
        logger.setLevel(30      )

        with CaptureLogger(lowerCAmelCase_      ) as cap_logger:
               A_							:       Optional[Any]	      =				PNDMScheduler.from_pretrained(
                   '''hf-internal-testing/tiny-stable-diffusion-torch''' ,       subfolder='''scheduler'''      )

        assert pndm.__class__ == PNDMScheduler
        # no warning should be thrown
        assert cap_logger.out == ""
 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        A_							:       Union[str, Any]	      =				logging.get_logger('''diffusers.configuration_utils'''      )
        # 30 for warning
        logger.setLevel(30      )

        with CaptureLogger(lowerCAmelCase_      ) as cap_logger:
               A_							:       Dict	      =				DDPMScheduler.from_pretrained(
                   '''hf-internal-testing/tiny-stable-diffusion-torch''' ,       subfolder='''scheduler''' ,       prediction_type='''sample''' ,       beta_end=8 ,       )

        with CaptureLogger(lowerCAmelCase_      ) as cap_logger_a:
               A_							:       Optional[Any]	      =				DDPMScheduler.from_pretrained('''google/ddpm-celebahq-256''' ,       beta_start=88      )

        assert ddpm.__class__ == DDPMScheduler
        assert ddpm.config.prediction_type == "sample"
        assert ddpm.config.beta_end == 8
        assert ddpm_a.config.beta_start == 88

        # no warning should be thrown
        assert cap_logger.out == ""
        assert cap_logger_a.out == ""
 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        A_							:       Dict	      =				logging.get_logger('''diffusers.configuration_utils'''      )
        # 30 for warning
        logger.setLevel(30      )

        with CaptureLogger(lowerCAmelCase_      ) as cap_logger:
               A_							:       str	      =				DPMSolverMultistepScheduler.from_pretrained(
                   '''hf-internal-testing/tiny-stable-diffusion-torch''' ,       subfolder='''scheduler'''      )

        assert dpm.__class__ == DPMSolverMultistepScheduler
        # no warning should be thrown
        assert cap_logger.out == ""
 def  UpperCAmelCase   (    self      ):

        """simple docstring"""

        A_							:       Optional[int]	      =				SampleObject()

        A_							:       str	      =				{k: v for k, v in config.config.items() if not k.startswith('''_'''      )}

        # make sure that default config has all keys in `_use_default_values`
        assert set(config_dict.keys()      ) == set(config.config._use_default_values      )

        with tempfile.TemporaryDirectory() as tmpdirname:
               config.save_config(lowerCAmelCase_      )

               # now loading it with SampleObject2 should put f into `_use_default_values`
               A_							:       List[Any]	      =				SampleObjecta.from_config(lowerCAmelCase_      )

               assert "f" in config._use_default_values
               assert config.f == [1, 3]

        # now loading the config, should **NOT** use [1, 3] for `f`, but the default [1, 4] value
        # **BECAUSE** it is part of `config._use_default_values`
        A_							:       Any	      =				SampleObjecta.from_config(config.config      )
        assert new_config.f == [5, 4]

        config.config._use_default_values.pop()
        A_							:       Optional[int]	      =				SampleObjecta.from_config(config.config      )
        assert new_config_a.f == [1, 3]

        # Nevertheless "e" should still be correctly loaded to [1, 3] from SampleObject2 instead of defaulting to [1, 5]
        assert new_config_a.e == [1, 3]




'''simple docstring'''





from .pipeline_kandinskya_a import KandinskyVaaPipeline
from .pipeline_kandinsky2_2_controlnet import KandinskyVaaControlnetPipeline
from .pipeline_kandinsky2_2_controlnet_img2img import KandinskyVaaControlnetImgaImgPipeline
from .pipeline_kandinsky2_2_img2img import KandinskyVaaImgaImgPipeline
from .pipeline_kandinsky2_2_inpainting import KandinskyVaaInpaintPipeline
from .pipeline_kandinsky2_2_prior import KandinskyVaaPriorPipeline
from .pipeline_kandinsky2_2_prior_emb2emb import KandinskyVaaPriorEmbaEmbPipeline

