

import re
import tempfile
from pathlib import Path

import pytest
import yaml

from datasets.utils.readme import ReadMe


# @pytest.fixture
# def example_yaml_structure():

lowerCamelCase_    :		Dict							=	yaml.safe_load(
    """\\nname: \"\"\nallow_empty: false\nallow_empty_text: true\nsubsections:\n  - name: \"Dataset Card for X\" # First-level markdown heading\n    allow_empty: false\n    allow_empty_text: true\n    subsections:\n      - name: \"Table of Contents\"\n        allow_empty: false\n        allow_empty_text: false\n        subsections: null\n      - name: \"Dataset Description\"\n        allow_empty: false\n        allow_empty_text: false\n        subsections:\n          - name: \"Dataset Summary\"\n            allow_empty: false\n            allow_empty_text: false\n            subsections: null\n          - name: \"Supported Tasks and Leaderboards\"\n            allow_empty: true\n            allow_empty_text: true\n            subsections: null\n          - name: Languages\n            allow_empty: false\n            allow_empty_text: true\n            subsections: null\n"""
)


lowerCamelCase_    :		Any							=	{
    "name": "root",
    "text": "",
    "is_empty_text": True,
    "subsections": [
        {
            "name": "Dataset Card for My Dataset",
            "text": "",
            "is_empty_text": True,
            "subsections": [
                {"name": "Table of Contents", "text": "Some text here.", "is_empty_text": False, "subsections": []},
                {
                    "name": "Dataset Description",
                    "text": "Some text here.",
                    "is_empty_text": False,
                    "subsections": [
                        {
                            "name": "Dataset Summary",
                            "text": "Some text here.",
                            "is_empty_text": False,
                            "subsections": [],
                        },
                        {
                            "name": "Supported Tasks and Leaderboards",
                            "text": "",
                            "is_empty_text": True,
                            "subsections": [],
                        },
                        {"name": "Languages", "text": "Language Text", "is_empty_text": False, "subsections": []},
                    ],
                },
            ],
        }
    ],
}


lowerCamelCase_    :		List[str]							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"


lowerCamelCase_    :		List[Any]							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n#### Extra Ignored Subsection\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

lowerCamelCase_    :		int							=	{
    "name": "root",
    "text": "",
    "is_empty_text": True,
    "subsections": [
        {
            "name": "Dataset Card for My Dataset",
            "text": "",
            "is_empty_text": True,
            "subsections": [
                {"name": "Table of Contents", "text": "Some text here.", "is_empty_text": False, "subsections": []},
                {
                    "name": "Dataset Description",
                    "text": "Some text here.",
                    "is_empty_text": False,
                    "subsections": [
                        {
                            "name": "Dataset Summary",
                            "text": "Some text here.",
                            "is_empty_text": False,
                            "subsections": [
                                {
                                    "name": "Extra Ignored Subsection",
                                    "text": "",
                                    "is_empty_text": True,
                                    "subsections": [],
                                }
                            ],
                        },
                        {
                            "name": "Supported Tasks and Leaderboards",
                            "text": "",
                            "is_empty_text": True,
                            "subsections": [],
                        },
                        {"name": "Languages", "text": "Language Text", "is_empty_text": False, "subsections": []},
                    ],
                },
            ],
        }
    ],
}

lowerCamelCase_    :		Tuple							=	"\\n---\n---\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

lowerCamelCase_    :		int							=	(
    "The following issues were found for the README at `{path}`:\n-\tEmpty YAML markers are present in the README."
)

lowerCamelCase_    :		int							=	"\\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

lowerCamelCase_    :		str							=	(
    "The following issues were found for the README at `{path}`:\n-\tNo YAML markers are present in the README."
)

lowerCamelCase_    :		Optional[Any]							=	"\\n---\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

lowerCamelCase_    :		List[str]							=	"The following issues were found for the README at `{path}`:\n-\tOnly the start of YAML tags present in the README."

lowerCamelCase_    :		Tuple							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"
lowerCamelCase_    :		Optional[Any]							=	"The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Dataset Summary` but it is empty.\n-\tExpected some text in section `Dataset Summary` but it is empty (text in subsections are ignored)."


lowerCamelCase_    :		Optional[int]							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n"
lowerCamelCase_    :		Dict							=	"The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Dataset Card for My Dataset` but it is empty.\n-\tSection `Dataset Card for My Dataset` expected the following subsections: `Table of Contents`, `Dataset Description`. Found 'None'."

lowerCamelCase_    :		Optional[int]							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Languages\nLanguage Text\n"

lowerCamelCase_    :		Tuple							=	"The following issues were found for the README at `{path}`:\n-\tSection `Dataset Description` is missing subsection: `Supported Tasks and Leaderboards`."


lowerCamelCase_    :		Tuple							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\n"

lowerCamelCase_    :		List[str]							=	"The following issues were found for the README at `{path}`:\n-\tExpected some content in section `Languages` but it is empty."

lowerCamelCase_    :		Optional[int]							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"
lowerCamelCase_    :		Optional[Any]							=	"The following issues were found for the README at `{path}`:\n-\tThe README has no first-level headings. One heading is expected. Skipping further validation for this README."

lowerCamelCase_    :		Optional[int]							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n# Dataset Card My Dataset\n"

lowerCamelCase_    :		int							=	"The following issues were found for the README at `{path}`:\n-\tThe README has several first-level headings: `Dataset Card for My Dataset`, `Dataset Card My Dataset`. Only one heading is expected. Skipping further validation for this README."

lowerCamelCase_    :		Optional[Any]							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

lowerCamelCase_    :		Tuple							=	"The following issues were found for the README at `{path}`:\n-\tNo first-level heading starting with `Dataset Card for` found in README. Skipping further validation for this README."

lowerCamelCase_    :		Any							=	""

lowerCamelCase_    :		Optional[int]							=	"The following issues were found for the README at `{path}`:\n-\tThe README has no first-level headings. One heading is expected. Skipping further validation for this README.\n-\tNo YAML markers are present in the README."

lowerCamelCase_    :		Optional[int]							=	"\\n---\nlanguage:\n- zh\n- en\n---\n\n# Dataset Card for My Dataset\n# Dataset Card for My Dataset\n## Table of Contents\nSome text here.\n## Dataset Description\nSome text here.\n### Dataset Summary\nSome text here.\n### Supported Tasks and Leaderboards\n### Languages\nLanguage Text\n"

lowerCamelCase_    :		List[str]							=	"The following issues were found while parsing the README at `{path}`:\n-\tMultiple sections with the same heading `Dataset Card for My Dataset` have been found. Please keep only one of these sections."
@pytest.mark.parametrize(
    'readme_md, expected_dict'				,    [
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ]				,    )
def 							lowerCAmelCase(     __lowerCamelCase				,    __lowerCamelCase						):
					assert ReadMe.from_string(lowerCAmelCase__				,    lowerCAmelCase__						).to_dict() == expected_dict
@pytest.mark.parametrize(
    'readme_md, expected_error'				,    [
        (README_NO_YAML, EXPECTED_ERROR_README_NO_YAML),
        (README_EMPTY_YAML, EXPECTED_ERROR_README_EMPTY_YAML),
        (README_INCORRECT_YAML, EXPECTED_ERROR_README_INCORRECT_YAML),
        (README_EMPTY, EXPECTED_ERROR_README_EMPTY),
        (README_NONE_SUBSECTION, EXPECTED_ERROR_README_NONE_SUBSECTION),
        (README_MISSING_FIRST_LEVEL, EXPECTED_ERROR_README_MISSING_FIRST_LEVEL),
        (README_MISSING_SUBSECTION, EXPECTED_ERROR_README_MISSING_SUBSECTION),
        (README_MISSING_TEXT, EXPECTED_ERROR_README_MISSING_TEXT),
        (README_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_WRONG_FIRST_LEVEL),
        (README_MULTIPLE_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_MULTIPLE_WRONG_FIRST_LEVEL),
        (README_MISSING_CONTENT, EXPECTED_ERROR_README_MISSING_CONTENT),
    ]				,    )
def 							lowerCAmelCase(     __lowerCamelCase				,    __lowerCamelCase						):



					with pytest.raises(lowerCAmelCase__				,    match=re.escape(expected_error.format(path='root'						)						)						):
										__a		 =       ReadMe.from_string(lowerCAmelCase__				,    lowerCAmelCase__						)
										readme.validate()
@pytest.mark.parametrize(
    'readme_md, expected_error'				,    [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ]				,    )
def 							lowerCAmelCase(     __lowerCamelCase				,    __lowerCamelCase						):



					with pytest.raises(lowerCAmelCase__				,    match=re.escape(expected_error.format(path='root'						)						)						):
										ReadMe.from_string(lowerCAmelCase__				,    lowerCAmelCase__						)
@pytest.mark.parametrize(
    'readme_md,'				,    [
        (README_MULTIPLE_SAME_HEADING_1),
    ]				,    )
def 							lowerCAmelCase(     __lowerCamelCase						):
					ReadMe.from_string(lowerCAmelCase__				,    lowerCAmelCase__				,    suppress_parsing_errors=lowerCAmelCase__						)
@pytest.mark.parametrize(
    'readme_md, expected_dict'				,    [
        (README_CORRECT, CORRECT_DICT),
        (README_CORRECT_FOUR_LEVEL, CORRECT_DICT_FOUR_LEVEL),
    ]				,    )
def 							lowerCAmelCase(     __lowerCamelCase				,    __lowerCamelCase						):



					with tempfile.TemporaryDirectory() as tmp_dir:
										__a		 =       Path(lowerCAmelCase__						) / 'README.md'
										with open(lowerCAmelCase__				,    'w+'						) as readme_file:
															readme_file.write(lowerCAmelCase__						)
										__a		 =       ReadMe.from_readme(lowerCAmelCase__				,    lowerCAmelCase__						).to_dict()
										assert out["name"] == path
										assert out["text"] == ""
										assert out["is_empty_text"]
										assert out["subsections"] == expected_dict["subsections"]
@pytest.mark.parametrize(
    'readme_md, expected_error'				,    [
        (README_NO_YAML, EXPECTED_ERROR_README_NO_YAML),
        (README_EMPTY_YAML, EXPECTED_ERROR_README_EMPTY_YAML),
        (README_INCORRECT_YAML, EXPECTED_ERROR_README_INCORRECT_YAML),
        (README_EMPTY, EXPECTED_ERROR_README_EMPTY),
        (README_NONE_SUBSECTION, EXPECTED_ERROR_README_NONE_SUBSECTION),
        (README_MISSING_FIRST_LEVEL, EXPECTED_ERROR_README_MISSING_FIRST_LEVEL),
        (README_MISSING_SUBSECTION, EXPECTED_ERROR_README_MISSING_SUBSECTION),
        (README_MISSING_TEXT, EXPECTED_ERROR_README_MISSING_TEXT),
        (README_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_WRONG_FIRST_LEVEL),
        (README_MULTIPLE_WRONG_FIRST_LEVEL, EXPECTED_ERROR_README_MULTIPLE_WRONG_FIRST_LEVEL),
        (README_MISSING_CONTENT, EXPECTED_ERROR_README_MISSING_CONTENT),
    ]				,    )
def 							lowerCAmelCase(     __lowerCamelCase				,    __lowerCamelCase						):



					with tempfile.TemporaryDirectory() as tmp_dir:
										__a		 =       Path(lowerCAmelCase__						) / 'README.md'
										with open(lowerCAmelCase__				,    'w+'						) as readme_file:
															readme_file.write(lowerCAmelCase__						)
										__a		 =       expected_error.format(path=lowerCAmelCase__						)
										with pytest.raises(lowerCAmelCase__				,    match=re.escape(lowerCAmelCase__						)						):
															__a		 =       ReadMe.from_readme(lowerCAmelCase__				,    lowerCAmelCase__						)
															readme.validate()
@pytest.mark.parametrize(
    'readme_md, expected_error'				,    [
        (README_MULTIPLE_SAME_HEADING_1, EXPECTED_ERROR_README_MULTIPLE_SAME_HEADING_1),
    ]				,    )
def 							lowerCAmelCase(     __lowerCamelCase				,    __lowerCamelCase						):



					with tempfile.TemporaryDirectory() as tmp_dir:
										__a		 =       Path(lowerCAmelCase__						) / 'README.md'
										with open(lowerCAmelCase__				,    'w+'						) as readme_file:
															readme_file.write(lowerCAmelCase__						)
										__a		 =       expected_error.format(path=lowerCAmelCase__						)
										with pytest.raises(lowerCAmelCase__				,    match=re.escape(lowerCAmelCase__						)						):
															ReadMe.from_readme(lowerCAmelCase__				,    lowerCAmelCase__						)



@pytest.mark.parametrize(
    'readme_md,'				,    [
        (README_MULTIPLE_SAME_HEADING_1),
    ]				,    )
def 							lowerCAmelCase(     __lowerCamelCase						):



					with tempfile.TemporaryDirectory() as tmp_dir:
										__a		 =       Path(lowerCAmelCase__						) / 'README.md'
										with open(lowerCAmelCase__				,    'w+'						) as readme_file:
															readme_file.write(lowerCAmelCase__						)
										ReadMe.from_readme(lowerCAmelCase__				,    lowerCAmelCase__				,    suppress_parsing_errors=lowerCAmelCase__						)





import tempfile

import numpy as np
import torch
from transformers import AutoTokenizer, TaEncoderModel

from diffusers import DDPMScheduler, UNetaDConditionModel
from diffusers.models.attention_processor import AttnAddedKVProcessor
from diffusers.pipelines.deepfloyd_if import IFWatermarker
from diffusers.utils.testing_utils import torch_device

from ..test_pipelines_common import to_np


class 						a__							:



			def __SCREAMING_SNAKE_CASE							(						self   )			->   Optional[Any]:
								torch.manual_seed(0   )
								__a		 =       TaEncoderModel.from_pretrained('hf-internal-testing/tiny-random-t5'   )

								torch.manual_seed(0   )
								__a		 =       AutoTokenizer.from_pretrained('hf-internal-testing/tiny-random-t5'   )

								torch.manual_seed(0   )
								__a		 =       UNetaDConditionModel(
								    sample_size=3_2				,						layers_per_block=1				,						block_out_channels=[3_2, 6_4]				,						down_block_types=[
								        'ResnetDownsampleBlock2D',
								        'SimpleCrossAttnDownBlock2D',
								    ]				,						mid_block_type='UNetMidBlock2DSimpleCrossAttn'				,						up_block_types=['SimpleCrossAttnUpBlock2D', 'ResnetUpsampleBlock2D']				,						in_channels=3				,						out_channels=6				,						cross_attention_dim=3_2				,						encoder_hid_dim=3_2				,						attention_head_dim=8				,						addition_embed_type='text'				,						addition_embed_type_num_heads=2				,						cross_attention_norm='group_norm'				,						resnet_time_scale_shift='scale_shift'				,						act_fn='gelu'				,						)
								unet.set_attn_processor(AttnAddedKVProcessor()   )  # For reproducibility tests

								torch.manual_seed(0   )
								__a		 =       DDPMScheduler(
								    num_train_timesteps=1_0_0_0				,						beta_schedule='squaredcos_cap_v2'				,						beta_start=0.0_001				,						beta_end=0.02				,						thresholding=UpperCAmelCase				,						dynamic_thresholding_ratio=0.95				,						sample_max_value=1.0				,						prediction_type='epsilon'				,						variance_type='learned_range'				,						)

								torch.manual_seed(0   )
								__a		 =       IFWatermarker()

								return {
								    "text_encoder": text_encoder,
								    "tokenizer": tokenizer,
								    "unet": unet,
								    "scheduler": scheduler,
								    "watermarker": watermarker,
								    "safety_checker": None,
								    "feature_extractor": None,
								}



			def __SCREAMING_SNAKE_CASE							(						self   )			->   Optional[int]:
								torch.manual_seed(0   )
								__a		 =       TaEncoderModel.from_pretrained('hf-internal-testing/tiny-random-t5'   )

								torch.manual_seed(0   )
								__a		 =       AutoTokenizer.from_pretrained('hf-internal-testing/tiny-random-t5'   )

								torch.manual_seed(0   )
								__a		 =       UNetaDConditionModel(
								    sample_size=3_2				,						layers_per_block=[1, 2]				,						block_out_channels=[3_2, 6_4]				,						down_block_types=[
								        'ResnetDownsampleBlock2D',
								        'SimpleCrossAttnDownBlock2D',
								    ]				,						mid_block_type='UNetMidBlock2DSimpleCrossAttn'				,						up_block_types=['SimpleCrossAttnUpBlock2D', 'ResnetUpsampleBlock2D']				,						in_channels=6				,						out_channels=6				,						cross_attention_dim=3_2				,						encoder_hid_dim=3_2				,						attention_head_dim=8				,						addition_embed_type='text'				,						addition_embed_type_num_heads=2				,						cross_attention_norm='group_norm'				,						resnet_time_scale_shift='scale_shift'				,						act_fn='gelu'				,						class_embed_type='timestep'				,						mid_block_scale_factor=1.414				,						time_embedding_act_fn='gelu'				,						time_embedding_dim=3_2				,						)
								unet.set_attn_processor(AttnAddedKVProcessor()   )  # For reproducibility tests

								torch.manual_seed(0   )
								__a		 =       DDPMScheduler(
								    num_train_timesteps=1_0_0_0				,						beta_schedule='squaredcos_cap_v2'				,						beta_start=0.0_001				,						beta_end=0.02				,						thresholding=UpperCAmelCase				,						dynamic_thresholding_ratio=0.95				,						sample_max_value=1.0				,						prediction_type='epsilon'				,						variance_type='learned_range'				,						)

								torch.manual_seed(0   )
								__a		 =       DDPMScheduler(
								    num_train_timesteps=1_0_0_0				,						beta_schedule='squaredcos_cap_v2'				,						beta_start=0.0_001				,						beta_end=0.02				,						)

								torch.manual_seed(0   )
								__a		 =       IFWatermarker()

								return {
								    "text_encoder": text_encoder,
								    "tokenizer": tokenizer,
								    "unet": unet,
								    "scheduler": scheduler,
								    "image_noising_scheduler": image_noising_scheduler,
								    "watermarker": watermarker,
								    "safety_checker": None,
								    "feature_extractor": None,
								}



			def __SCREAMING_SNAKE_CASE							(						self   )			->   str:
								__a		 =       self.get_dummy_components()
								__a		 =       self.pipeline_class(**UpperCAmelCase   )
								pipe.to(UpperCAmelCase   )
								pipe.set_progress_bar_config(disable=UpperCAmelCase   )

								__a		 =       self.get_dummy_inputs(UpperCAmelCase   )

								__a		 =       inputs['prompt']
								__a		 =       inputs['generator']
								__a		 =       inputs['num_inference_steps']
								__a		 =       inputs['output_type']

								if "image" in inputs:
													__a		 =       inputs['image']
								else:
													__a		 =       None

								if "mask_image" in inputs:
													__a		 =       inputs['mask_image']
								else:
													__a		 =       None

								if "original_image" in inputs:
													__a		 =       inputs['original_image']
								else:
													__a		 =       None

								__a ,				__a		 =       pipe.encode_prompt(UpperCAmelCase   )

								# inputs with prompt converted to embeddings
								__a		 =       {
								    'prompt_embeds': prompt_embeds,
								    'negative_prompt_embeds': negative_prompt_embeds,
								    'generator': generator,
								    'num_inference_steps': num_inference_steps,
								    'output_type': output_type,
								}

								if image is not None:
													__a		 =       image

								if mask_image is not None:
													__a		 =       mask_image

								if original_image is not None:
													__a		 =       original_image

								# set all optional components to None
								for optional_component in pipe._optional_components:
													setattr(UpperCAmelCase				,						UpperCAmelCase				,						UpperCAmelCase   )

								__a		 =       pipe(**UpperCAmelCase   )[0]

								with tempfile.TemporaryDirectory() as tmpdir:
													pipe.save_pretrained(UpperCAmelCase   )
													__a		 =       self.pipeline_class.from_pretrained(UpperCAmelCase   )
													pipe_loaded.to(UpperCAmelCase   )
													pipe_loaded.set_progress_bar_config(disable=UpperCAmelCase   )

								pipe_loaded.unet.set_attn_processor(AttnAddedKVProcessor()   )  # For reproducibility tests

								for optional_component in pipe._optional_components:
													self.assertTrue(
													    getattr(UpperCAmelCase				,						UpperCAmelCase   ) is None				,						f'''`{optional_component}` did not stay set to None after loading.'''				,						)

								__a		 =       self.get_dummy_inputs(UpperCAmelCase   )

								__a		 =       inputs['generator']
								__a		 =       inputs['num_inference_steps']
								__a		 =       inputs['output_type']

								# inputs with prompt converted to embeddings
								__a		 =       {
								    'prompt_embeds': prompt_embeds,
								    'negative_prompt_embeds': negative_prompt_embeds,
								    'generator': generator,
								    'num_inference_steps': num_inference_steps,
								    'output_type': output_type,
								}

								if image is not None:
													__a		 =       image

								if mask_image is not None:
													__a		 =       mask_image

								if original_image is not None:
													__a		 =       original_image

								__a		 =       pipe_loaded(**UpperCAmelCase   )[0]

								__a		 =       np.abs(to_np(UpperCAmelCase   ) - to_np(UpperCAmelCase   )   ).max()
								self.assertLess(UpperCAmelCase				,						1e-4   )







			def __SCREAMING_SNAKE_CASE							(						self   )			->   int:
								__a		 =       self.get_dummy_components()
								__a		 =       self.pipeline_class(**UpperCAmelCase   )
								pipe.to(UpperCAmelCase   )
								pipe.set_progress_bar_config(disable=UpperCAmelCase   )

								__a		 =       self.get_dummy_inputs(UpperCAmelCase   )
								__a		 =       pipe(**UpperCAmelCase   )[0]

								with tempfile.TemporaryDirectory() as tmpdir:
													pipe.save_pretrained(UpperCAmelCase   )
													__a		 =       self.pipeline_class.from_pretrained(UpperCAmelCase   )
													pipe_loaded.to(UpperCAmelCase   )
													pipe_loaded.set_progress_bar_config(disable=UpperCAmelCase   )

								pipe_loaded.unet.set_attn_processor(AttnAddedKVProcessor()   )  # For reproducibility tests

								__a		 =       self.get_dummy_inputs(UpperCAmelCase   )
								__a		 =       pipe_loaded(**UpperCAmelCase   )[0]

								__a		 =       np.abs(to_np(UpperCAmelCase   ) - to_np(UpperCAmelCase   )   ).max()
								self.assertLess(UpperCAmelCase				,						1e-4   )




