




import gc
import unittest

import numpy as np
import torch
from transformers import CLIPTextConfig, CLIPTextModelWithProjection, CLIPTokenizer

from diffusers import HeunDiscreteScheduler, PriorTransformer, ShapEPipeline
from diffusers.pipelines.shap_e import ShapERenderer
from diffusers.utils import load_numpy, slow
from diffusers.utils.testing_utils import require_torch_gpu, torch_device

from ..test_pipelines_common import PipelineTesterMixin, assert_mean_pixel_difference





class     lowerCamelCase__(					__lowerCamelCase ,     unittest.TestCase):
      UpperCAmelCase__      :	Dict    		=   ShapEPipeline
      UpperCAmelCase__      :	Any    		=   ['prompt']
      UpperCAmelCase__      :	List[Any]    		=   ['prompt']
      UpperCAmelCase__      :	str    		=   [
          'num_images_per_prompt',
          'num_inference_steps',
          'generator',
          'latents',
          'guidance_scale',
          'frame_size',
          'output_type',
          'return_dict',
      ]
      UpperCAmelCase__      :	Dict    		=   False
      @property
      def 			lowerCAmelCase__						(				self: int   ):
             return 32
      @property
      def 			lowerCAmelCase__						(				self: Union[str, Any]   ):
             return 32
      @property
      def 			lowerCAmelCase__						(				self: int   ):
             return self.time_input_dim * 4
      @property
      def 			lowerCAmelCase__						(				self: Optional[Any]   ):
             return 8
      @property
      def 			lowerCAmelCase__						(				self: List[str]   ):
             __lowerCamelCase   =						CLIPTokenizer.from_pretrained("""hf-internal-testing/tiny-random-clip"""   )
             return tokenizer
      @property
      def 			lowerCAmelCase__						(				self: str   ):
             torch.manual_seed(0   )
             __lowerCamelCase   =						CLIPTextConfig(
                 bos_token_id=0    ,       eos_token_id=2    ,       hidden_size=self.text_embedder_hidden_size    ,       projection_dim=self.text_embedder_hidden_size    ,       intermediate_size=37    ,       layer_norm_eps=1E-05    ,       num_attention_heads=4    ,       num_hidden_layers=5    ,       pad_token_id=1    ,       vocab_size=10_00    ,       )
             return CLIPTextModelWithProjection(UpperCamelCase_   )
      @property
      def 			lowerCAmelCase__						(				self: List[str]   ):
             torch.manual_seed(0   )

             __lowerCamelCase   =						{
                 """num_attention_heads""": 2,
                 """attention_head_dim""": 16,
                 """embedding_dim""": self.time_input_dim,
                 """num_embeddings""": 32,
                 """embedding_proj_dim""": self.text_embedder_hidden_size,
                 """time_embed_dim""": self.time_embed_dim,
                 """num_layers""": 1,
                 """clip_embed_dim""": self.time_input_dim * 2,
                 """additional_embeddings""": 0,
                 """time_embed_act_fn""": """gelu""",
                 """norm_in_type""": """layer""",
                 """encoder_hid_proj_type""": None,
                 """added_emb_type""": None,
             }

             __lowerCamelCase   =						PriorTransformer(**UpperCamelCase_   )
             return model
      @property
      def 			lowerCAmelCase__						(				self: Dict   ):
             torch.manual_seed(0   )

             __lowerCamelCase   =						{
                 """param_shapes""": (
                     (self.renderer_dim, 93),
                     (self.renderer_dim, 8),
                     (self.renderer_dim, 8),
                     (self.renderer_dim, 8),
                 ),
                 """d_latent""": self.time_input_dim,
                 """d_hidden""": self.renderer_dim,
                 """n_output""": 12,
                 """background""": (
                     0.1,
                     0.1,
                     0.1,
                 ),
             }
             __lowerCamelCase   =						ShapERenderer(**UpperCamelCase_   )
             return model
      def 			lowerCAmelCase__						(				self: str   ):
             __lowerCamelCase   =						self.dummy_prior
             __lowerCamelCase   =						self.dummy_text_encoder
             __lowerCamelCase   =						self.dummy_tokenizer
             __lowerCamelCase   =						self.dummy_renderer

             __lowerCamelCase   =						HeunDiscreteScheduler(
                 beta_schedule="""exp"""    ,       num_train_timesteps=10_24    ,       prediction_type="""sample"""    ,       use_karras_sigmas=UpperCamelCase_    ,       clip_sample=UpperCamelCase_    ,       clip_sample_range=1.0    ,       )
             __lowerCamelCase   =						{
                 """prior""": prior,
                 """text_encoder""": text_encoder,
                 """tokenizer""": tokenizer,
                 """renderer""": renderer,
                 """scheduler""": scheduler,
             }

             return components
      def 			lowerCAmelCase__						(				self: str    ,       UpperCamelCase_: Dict    ,       UpperCamelCase_: Optional[Any]=0   ):
             if str(UpperCamelCase_   ).startswith("""mps"""   ):
                    __lowerCamelCase   =						torch.manual_seed(UpperCamelCase_   )
             else:
                    __lowerCamelCase   =						torch.Generator(device=UpperCamelCase_   ).manual_seed(UpperCamelCase_   )
             __lowerCamelCase   =						{
                 """prompt""": """horse""",
                 """generator""": generator,
                 """num_inference_steps""": 1,
                 """frame_size""": 32,
                 """output_type""": """np""",
             }
             return inputs
      def 			lowerCAmelCase__						(				self: List[str]   ):
             __lowerCamelCase   =						"""cpu"""

             __lowerCamelCase   =						self.get_dummy_components()

             __lowerCamelCase   =						self.pipeline_class(**UpperCamelCase_   )
             __lowerCamelCase   =						pipe.to(UpperCamelCase_   )

             pipe.set_progress_bar_config(disable=UpperCamelCase_   )

             __lowerCamelCase   =						pipe(**self.get_dummy_inputs(UpperCamelCase_   )   )
             __lowerCamelCase   =						output.images[0]
             __lowerCamelCase   =						image[0, -3:, -3:, -1]

             assert image.shape == (20, 32, 32, 3)

             __lowerCamelCase   =						np.array(
                 [
                     0.0003_9216,
                     0.0003_9216,
                     0.0003_9216,
                     0.0003_9216,
                     0.0003_9216,
                     0.0003_9216,
                     0.0003_9216,
                     0.0003_9216,
                     0.0003_9216,
                 ]   )

             assert np.abs(image_slice.flatten() - expected_slice   ).max() < 1E-2
      def 			lowerCAmelCase__						(				self: List[str]   ):
             # NOTE: Larger batch sizes cause this test to timeout, only test on smaller batches
             self._test_inference_batch_consistent(batch_sizes=[1, 2]   )
      def 			lowerCAmelCase__						(				self: Optional[int]   ):
             __lowerCamelCase   =						torch_device == """cpu"""
             __lowerCamelCase   =						True

             self._test_inference_batch_single_identical(
                 batch_size=2    ,       test_max_difference=UpperCamelCase_    ,       relax_max_difference=UpperCamelCase_    ,       )




      def 			lowerCAmelCase__						(				self: Union[str, Any]   ):
             __lowerCamelCase   =						self.get_dummy_components()
             __lowerCamelCase   =						self.pipeline_class(**UpperCamelCase_   )
             __lowerCamelCase   =						pipe.to(UpperCamelCase_   )
             pipe.set_progress_bar_config(disable=UpperCamelCase_   )

             __lowerCamelCase   =						1
             __lowerCamelCase   =						2

             __lowerCamelCase   =						self.get_dummy_inputs(UpperCamelCase_   )

             for key in inputs.keys():
                    if key in self.batch_params:
                           __lowerCamelCase   =						batch_size * [inputs[key]]

             __lowerCamelCase   =						pipe(**UpperCamelCase_    ,       num_images_per_prompt=UpperCamelCase_   )[0]

             assert images.shape[0] == batch_size * num_images_per_prompt





@slow
@require_torch_gpu
class     lowerCamelCase__(					unittest.TestCase):
      def 			lowerCAmelCase__						(				self: List[Any]   ):
             # clean up the VRAM after each test
             super().tearDown()
             gc.collect()
             torch.cuda.empty_cache()




      def 			lowerCAmelCase__						(				self: Tuple   ):
             __lowerCamelCase   =						load_numpy(
                 """https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main"""
                 """/shap_e/test_shap_e_np_out.npy"""   )
             __lowerCamelCase   =						ShapEPipeline.from_pretrained("""openai/shap-e"""   )
             __lowerCamelCase   =						pipe.to(UpperCamelCase_   )
             pipe.set_progress_bar_config(disable=UpperCamelCase_   )

             __lowerCamelCase   =						torch.Generator(device=UpperCamelCase_   ).manual_seed(0   )

             __lowerCamelCase   =						pipe(
                 """a shark"""    ,       generator=UpperCamelCase_    ,       guidance_scale=15.0    ,       num_inference_steps=64    ,       frame_size=64    ,       output_type="""np"""    ,       ).images[0]

             assert images.shape == (20, 64, 64, 3)

             assert_mean_pixel_difference(UpperCamelCase_    ,       UpperCamelCase_   )







import json
import os
from datetime import date
from pathlib import Path

from tabulate import DataRow, TableFormat, tabulate


UpperCAmelCase_									=	TableFormat(
    lineabove=None,
    linebelowheader=None,
    linebetweenrows=None,
    linebelow=None,
    headerrow=DataRow('', '|', '|'),
    datarow=DataRow('', '|', '|'),
    padding=1,
    with_header_hide=None,
)


UpperCAmelCase_									=	[]
UpperCAmelCase_									=	[]

UpperCAmelCase_									=	{'type': 'section', 'text': {'type': 'plain_text', 'text': 'No failed tests! 🤗', 'emoji': True}}

UpperCAmelCase_									=	[
    {
        'type': 'header',
        'text': {
            'type': 'plain_text',
            'text': f"""🤗 Accelerate nightly {os.environ.get("TEST_TYPE", "")} test results""",
            'emoji': True,
        },
    }
]

UpperCAmelCase_									=	0
for log in Path().glob('*.log'):
       UpperCAmelCase_									=	0
       with open(log, 'r') as f:
              for line in f:
                     UpperCAmelCase_									=	json.loads(line)
                     if line.get('nodeid', '') != "":
                            UpperCAmelCase_									=	line['nodeid']
                            if line.get('duration', None) is not None:
                                   UpperCAmelCase_									=	f"""{line["duration"]:.4f}"""
                                   if line.get('outcome', '') == "failed":
                                          section_num_failed += 1
                                          failed.append([test, duration, log.name.split('_')[0]])
                                          total_num_failed += 1
       group_info.append([str(log), section_num_failed, failed])
       UpperCAmelCase_									=	[]
       log.unlink()

UpperCAmelCase_									=	''
UpperCAmelCase_									=	[]
if total_num_failed > 0:
       for name, num_failed, failed_tests in group_info:
              if num_failed > 0:
                     if num_failed == 1:
                            message += f"*{name[1:]}: {num_failed} failed test*\n"
                     else:
                            message += f"*{name[1:]}: {num_failed} failed tests*\n"
                     UpperCAmelCase_									=	[]
                     UpperCAmelCase_									=	{}
                     for test in failed_tests:
                            UpperCAmelCase_									=	test[0].split('::')
                            UpperCAmelCase_									=	data[0].split('/')[-1]
                            if data[0] not in filesafailed:
                                   UpperCAmelCase_									=	[data[1:]]
                            else:
                                   filesafailed[data[0]] += [data[1:]]
                            failed_table.append(data)

                     UpperCAmelCase_									=	[test[0] for test in failed_table]
                     UpperCAmelCase_									=	list(set(files))
                     # Count number of instances in failed_tests
                     UpperCAmelCase_									=	[]
                     for file in individual_files:
                            table.append([file, len(filesafailed[file])])

                     UpperCAmelCase_									=	tabulate(
                         table,
                         headers=['Test Location', 'Num Failed'],
                         tablefmt=hf_table_format,
                         stralign='right',
                     )
                     message += f"\n```\n{failed_table}\n```"
                     all_filesafailed.append(filesafailed)
       if len(message) > 3_000:
              UpperCAmelCase_									=	'Too many failed tests, please see the full report in the Action results.'
              UpperCAmelCase_									=	len(err) + 10
              UpperCAmelCase_									=	message[: 3_000 - offset] + f"""\n...\n```\n{err}"""
       print(f"""### {message}""")
else:
       UpperCAmelCase_									=	'No failed tests! 🤗'
       print(f"""## {message}""")
       payload.append(no_error_payload)

if os.environ.get('TEST_TYPE', '') != "":
       from slack_sdk import WebClient

       UpperCAmelCase_									=	WebClient(token=os.environ['SLACK_API_TOKEN'])
       if message != "No failed tests! 🤗":
              UpperCAmelCase_									=	{
                  'type': 'section',
                  'text': {
                      'type': 'mrkdwn',
                      'text': message,
                  },
              }
              payload.append(md_report)
              UpperCAmelCase_									=	{
                  'type': 'section',
                  'text': {
                      'type': 'mrkdwn',
                      'text': '*For more details:*',
                  },
                  'accessory': {
                      'type': 'button',
                      'text': {
                          'type': 'plain_text',
                          'text': 'Check Action results',
                          'emoji': True,
                      },
                      'url': f"""https://github.com/{os.environ["GITHUB_REPOSITORY"]}/actions/runs/{os.environ["GITHUB_RUN_ID"]}""",
                  },
              }
              payload.append(action_button)
              UpperCAmelCase_									=	{
                  'type': 'context',
                  'elements': [
                      {
                          'type': 'plain_text',
                          'text': f"""Nightly {os.environ.get("TEST_TYPE")} test results for {date.today()}""",
                      }
                  ],
              }
              payload.append(date_report)
       UpperCAmelCase_									=	client.chat_postMessage(channel='#accelerate-ci-daily', text=message, blocks=payload)
       UpperCAmelCase_									=	response.data['ts']
       for failed_file in all_filesafailed:
              for test_location, test_failures in failed_file.items():
                     # Keep only the first instance of the test name
                     UpperCAmelCase_									=	''
                     for i, row in enumerate(test_failures):
                            if row[0] != test_class:
                                   UpperCAmelCase_									=	row[0]
                            else:
                                   UpperCAmelCase_									=	''

                     UpperCAmelCase_									=	{
                         'type': 'section',
                         'text': {
                             'type': 'mrkdwn',
                             'text': f"""Test location: {test_location}\n```\n{tabulate(test_failures, headers=["Class", "Test"], tablefmt=hf_table_format, stralign="right")}\n```""",
                         },
                     }

                     client.chat_postMessage(
                         channel='#accelerate-ci-daily',
                         thread_ts=ts,
                         blocks=[payload],
                     )

