





'''simple docstring'''




import unittest

from diffusers.models.unet_ad_blocks import *  # noqa F403
from diffusers.utils import torch_device

from .test_unet_blocks_common import UNetBlockTesterMixin







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Tuple           =							DownBlockaD  # noqa F405
 _lowerCamelCase  : Optional[int]           =							"""down"""




 def      lowercase	(					self       :    List[Any]    ):
      _UpperCAmelCase     							=   [-0.0_2_3_2, -0.9_8_6_9, 0.8_0_5_4, -0.0_6_3_7, -0.1_6_8_8, -1.4_2_6_4, 0.4_4_7_0, -1.3_3_9_4, 0.0_9_0_4]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Tuple           =							ResnetDownsampleBlockaD  # noqa F405
 _lowerCamelCase  : Tuple           =							"""down"""




 def      lowercase	(					self       :    Dict    ):
      _UpperCAmelCase     							=   [0.0_7_1_0, 0.2_4_1_0, -0.7_3_2_0, -1.0_7_5_7, -1.1_3_4_3, 0.3_5_4_0, -0.0_1_3_3, -0.2_5_7_6, 0.0_9_4_8]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Optional[int]           =							AttnDownBlockaD  # noqa F405
 _lowerCamelCase  : Optional[int]           =							"""down"""




 def      lowercase	(					self       :    Dict    ):
      _UpperCAmelCase     							=   [0.0_6_3_6, 0.8_9_6_4, -0.6_2_3_4, -1.0_1_3_1, 0.0_8_4_4, 0.4_9_3_5, 0.3_4_3_7, 0.0_9_1_1, -0.2_9_5_7]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Optional[Any]           =							CrossAttnDownBlockaD  # noqa F405
 _lowerCamelCase  : Dict           =							"""down"""




 def      lowercase	(					self       :    List[str]    ):
      _UpperCAmelCase							,  _UpperCAmelCase     							=   super().prepare_init_args_and_inputs_for_common()
      _UpperCAmelCase     							=   3_2
      return init_dict, inputs_dict




 def      lowercase	(					self       :    List[str]    ):
      _UpperCAmelCase     							=   [0.2_2_3_8, -0.7_3_9_6, -0.2_2_5_5, -0.3_8_2_9, 0.1_9_2_5, 1.1_6_6_5, 0.0_6_0_3, -0.7_2_9_5, 0.1_9_8_3]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : str           =							SimpleCrossAttnDownBlockaD  # noqa F405
 _lowerCamelCase  : List[Any]           =							"""down"""




 @property
 def      lowercase	(					self       :    Optional[Any]    ):
      return super().get_dummy_input(include_encoder_hidden_states=snake_case_    )




 def      lowercase	(					self       :    Optional[int]    ):
      _UpperCAmelCase							,  _UpperCAmelCase     							=   super().prepare_init_args_and_inputs_for_common()
      _UpperCAmelCase     							=   3_2
      return init_dict, inputs_dict




 @unittest.skipIf(torch_device == "mps"	,			"MPS result is not consistent"    )
 def      lowercase	(					self       :    Dict    ):
      _UpperCAmelCase     							=   [0.7_9_2_1, -0.0_9_9_2, -0.1_9_6_2, -0.7_6_9_5, -0.4_2_4_2, 0.7_8_0_4, 0.4_7_3_7, 0.2_7_6_5, 0.3_3_3_8]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Union[str, Any]           =							SkipDownBlockaD  # noqa F405
 _lowerCamelCase  : Optional[int]           =							"""down"""




 @property
 def      lowercase	(					self       :    List[str]    ):
      return super().get_dummy_input(include_skip_sample=snake_case_    )




 def      lowercase	(					self       :    int    ):
      _UpperCAmelCase     							=   [-0.0_8_4_5, -0.2_0_8_7, -0.2_4_6_5, 0.0_9_7_1, 0.1_9_0_0, -0.0_4_8_4, 0.2_6_6_4, 0.4_1_7_9, 0.5_0_6_9]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : List[str]           =							AttnSkipDownBlockaD  # noqa F405
 _lowerCamelCase  : Optional[int]           =							"""down"""




 @property
 def      lowercase	(					self       :    Any    ):
      return super().get_dummy_input(include_skip_sample=snake_case_    )




 def      lowercase	(					self       :    Any    ):
      _UpperCAmelCase     							=   [0.5_5_3_9, 0.1_6_0_9, 0.4_9_2_4, 0.0_5_3_7, -0.1_9_9_5, 0.4_0_5_0, 0.0_9_7_9, -0.2_7_2_1, -0.0_6_4_2]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : str           =							DownEncoderBlockaD  # noqa F405
 _lowerCamelCase  : Any           =							"""down"""




 @property
 def      lowercase	(					self       :    Dict    ):
      return super().get_dummy_input(include_temb=snake_case_    )




 def      lowercase	(					self       :    List[str]    ):
      _UpperCAmelCase     							=   {
          "in_channels": 3_2,
          "out_channels": 3_2,
      }
      _UpperCAmelCase     							=   self.dummy_input
      return init_dict, inputs_dict




 def      lowercase	(					self       :    List[Any]    ):
      _UpperCAmelCase     							=   [1.1_1_0_2, 0.5_3_0_2, 0.4_8_7_2, -0.0_0_2_3, -0.8_0_4_2, 0.0_4_8_3, -0.3_4_8_9, -0.5_6_3_2, 0.7_6_2_6]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Optional[Any]           =							AttnDownEncoderBlockaD  # noqa F405
 _lowerCamelCase  : str           =							"""down"""




 @property
 def      lowercase	(					self       :    str    ):
      return super().get_dummy_input(include_temb=snake_case_    )




 def      lowercase	(					self       :    int    ):
      _UpperCAmelCase     							=   {
          "in_channels": 3_2,
          "out_channels": 3_2,
      }
      _UpperCAmelCase     							=   self.dummy_input
      return init_dict, inputs_dict




 def      lowercase	(					self       :    Tuple    ):
      _UpperCAmelCase     							=   [0.8_9_6_6, -0.1_4_8_6, 0.8_5_6_8, 0.8_1_4_1, -0.9_0_4_6, -0.1_3_4_2, -0.0_9_7_2, -0.7_4_1_7, 0.1_5_3_8]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : List[Any]           =							UNetMidBlockaD  # noqa F405
 _lowerCamelCase  : Dict           =							"""mid"""




 def      lowercase	(					self       :    List[str]    ):
      _UpperCAmelCase     							=   {
          "in_channels": 3_2,
          "temb_channels": 1_2_8,
      }
      _UpperCAmelCase     							=   self.dummy_input
      return init_dict, inputs_dict




 def      lowercase	(					self       :    List[Any]    ):
      _UpperCAmelCase     							=   [-0.1_0_6_2, 1.7_2_4_8, 0.3_4_9_4, 1.4_5_6_9, -0.0_9_1_0, -1.2_4_2_1, -0.9_9_8_4, 0.6_7_3_6, 1.0_0_2_8]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : List[Any]           =							UNetMidBlockaDCrossAttn  # noqa F405
 _lowerCamelCase  : Any           =							"""mid"""




 def      lowercase	(					self       :    Dict    ):
      _UpperCAmelCase							,  _UpperCAmelCase     							=   super().prepare_init_args_and_inputs_for_common()
      _UpperCAmelCase     							=   3_2
      return init_dict, inputs_dict




 def      lowercase	(					self       :    str    ):
      _UpperCAmelCase     							=   [0.0_1_8_7, 2.4_2_2_0, 0.4_4_8_4, 1.1_2_0_3, -0.6_1_2_1, -1.5_1_2_2, -0.8_2_7_0, 0.7_8_5_1, 1.8_3_3_5]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Dict           =							UNetMidBlockaDSimpleCrossAttn  # noqa F405
 _lowerCamelCase  : List[Any]           =							"""mid"""




 @property
 def      lowercase	(					self       :    Tuple    ):
      return super().get_dummy_input(include_encoder_hidden_states=snake_case_    )




 def      lowercase	(					self       :    Optional[int]    ):
      _UpperCAmelCase							,  _UpperCAmelCase     							=   super().prepare_init_args_and_inputs_for_common()
      _UpperCAmelCase     							=   3_2
      return init_dict, inputs_dict




 def      lowercase	(					self       :    Dict    ):
      _UpperCAmelCase     							=   [0.7_1_4_3, 1.9_9_7_4, 0.5_4_4_8, 1.3_9_7_7, 0.1_2_8_2, -1.1_2_3_7, -1.4_2_3_8, 0.5_5_3_0, 0.8_8_8_0]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : int           =							UpBlockaD  # noqa F405
 _lowerCamelCase  : Optional[int]           =							"""up"""




 @property
 def      lowercase	(					self       :    List[str]    ):
      return super().get_dummy_input(include_res_hidden_states_tuple=snake_case_    )




 def      lowercase	(					self       :    List[str]    ):
      _UpperCAmelCase     							=   [-0.2_0_4_1, -0.4_1_6_5, -0.3_0_2_2, 0.0_0_4_1, -0.6_6_2_8, -0.7_0_5_3, 0.1_9_2_8, -0.0_3_2_5, 0.0_5_2_3]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Any           =							ResnetUpsampleBlockaD  # noqa F405
 _lowerCamelCase  : Union[str, Any]           =							"""up"""




 @property
 def      lowercase	(					self       :    List[str]    ):
      return super().get_dummy_input(include_res_hidden_states_tuple=snake_case_    )




 def      lowercase	(					self       :    List[Any]    ):
      _UpperCAmelCase     							=   [0.2_2_8_7, 0.3_5_4_9, -0.1_3_4_6, 0.4_7_9_7, -0.1_7_1_5, -0.9_6_4_9, 0.7_3_0_5, -0.5_8_6_4, -0.6_2_4_4]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Any           =							CrossAttnUpBlockaD  # noqa F405
 _lowerCamelCase  : Any           =							"""up"""




 @property
 def      lowercase	(					self       :    List[str]    ):
      return super().get_dummy_input(include_res_hidden_states_tuple=snake_case_    )




 def      lowercase	(					self       :    List[Any]    ):
      _UpperCAmelCase							,  _UpperCAmelCase     							=   super().prepare_init_args_and_inputs_for_common()
      _UpperCAmelCase     							=   3_2
      return init_dict, inputs_dict




 def      lowercase	(					self       :    int    ):
      _UpperCAmelCase     							=   [-0.1_4_0_3, -0.3_5_1_5, -0.0_4_2_0, -0.1_4_2_5, 0.3_1_6_7, 0.5_0_9_4, -0.2_1_8_1, 0.5_9_3_1, 0.5_5_8_2]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Tuple           =							SimpleCrossAttnUpBlockaD  # noqa F405
 _lowerCamelCase  : Tuple           =							"""up"""




 @property
 def      lowercase	(					self       :    Any    ):
      return super().get_dummy_input(include_res_hidden_states_tuple=snake_case_	,			include_encoder_hidden_states=snake_case_    )




 def      lowercase	(					self       :    int    ):
      _UpperCAmelCase							,  _UpperCAmelCase     							=   super().prepare_init_args_and_inputs_for_common()
      _UpperCAmelCase     							=   3_2
      return init_dict, inputs_dict




 def      lowercase	(					self       :    List[str]    ):
      _UpperCAmelCase     							=   [0.2_6_4_5, 0.1_4_8_0, 0.0_9_0_9, 0.8_0_4_4, -0.9_7_5_8, -0.9_0_8_3, 0.0_9_9_4, -1.1_4_5_3, -0.7_4_0_2]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : List[str]           =							AttnUpBlockaD  # noqa F405
 _lowerCamelCase  : str           =							"""up"""




 @property
 def      lowercase	(					self       :    Optional[int]    ):
      return super().get_dummy_input(include_res_hidden_states_tuple=snake_case_    )




 @unittest.skipIf(torch_device == "mps"	,			"MPS result is not consistent"    )
 def      lowercase	(					self       :    Optional[int]    ):
      _UpperCAmelCase     							=   [0.0_9_7_9, 0.1_3_2_6, 0.0_0_2_1, 0.0_6_5_9, 0.2_2_4_9, 0.0_0_5_9, 0.1_1_3_2, 0.5_9_5_2, 0.1_0_3_3]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Dict           =							SkipUpBlockaD  # noqa F405
 _lowerCamelCase  : Union[str, Any]           =							"""up"""




 @property
 def      lowercase	(					self       :    Tuple    ):
      return super().get_dummy_input(include_res_hidden_states_tuple=snake_case_    )




 def      lowercase	(					self       :    str    ):
      _UpperCAmelCase     							=   [-0.0_8_9_3, -0.1_2_3_4, -0.1_5_0_6, -0.0_3_3_2, 0.0_1_2_3, -0.0_2_1_1, 0.0_5_6_6, 0.0_1_4_3, 0.0_3_6_2]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Optional[Any]           =							AttnSkipUpBlockaD  # noqa F405
 _lowerCamelCase  : int           =							"""up"""




 @property
 def      lowercase	(					self       :    List[Any]    ):
      return super().get_dummy_input(include_res_hidden_states_tuple=snake_case_    )




 def      lowercase	(					self       :    Optional[Any]    ):
      _UpperCAmelCase     							=   [0.0_3_6_1, 0.0_6_1_7, 0.2_7_8_7, -0.0_3_5_0, 0.0_3_4_2, 0.3_4_2_1, -0.0_8_4_3, 0.0_9_1_3, 0.3_0_1_5]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Dict           =							UpDecoderBlockaD  # noqa F405
 _lowerCamelCase  : Tuple           =							"""up"""




 @property
 def      lowercase	(					self       :    Optional[Any]    ):
      return super().get_dummy_input(include_temb=snake_case_    )




 def      lowercase	(					self       :    str    ):
      _UpperCAmelCase     							=   {"in_channels": 3_2, "out_channels": 3_2}

      _UpperCAmelCase     							=   self.dummy_input
      return init_dict, inputs_dict




 def      lowercase	(					self       :    Optional[int]    ):
      _UpperCAmelCase     							=   [0.4_4_0_4, 0.1_9_9_8, -0.9_8_8_6, -0.3_3_2_0, -0.3_1_2_8, -0.7_0_3_4, -0.6_9_5_5, -0.2_3_3_8, -0.3_1_3_7]
      super().test_output(snake_case_    )







class   A_	(  lowerCAmelCase_							,						unittest.TestCase			):
 _lowerCamelCase  : Union[str, Any]           =							AttnUpDecoderBlockaD  # noqa F405
 _lowerCamelCase  : int           =							"""up"""




 @property
 def      lowercase	(					self       :    Optional[int]    ):
      return super().get_dummy_input(include_temb=snake_case_    )




 def      lowercase	(					self       :    Optional[int]    ):
      _UpperCAmelCase     							=   {"in_channels": 3_2, "out_channels": 3_2}

      _UpperCAmelCase     							=   self.dummy_input
      return init_dict, inputs_dict




 def      lowercase	(					self       :    Optional[int]    ):
      _UpperCAmelCase     							=   [0.6_7_3_8, 0.4_4_9_1, 0.1_0_5_5, 1.0_7_1_0, 0.7_3_1_6, 0.3_3_3_9, 0.3_3_5_2, 0.1_0_2_3, 0.3_5_6_8]
      super().test_output(snake_case_    )







import argparse
import math
import traceback

import dateutil.parser as date_parser
import requests
def      a      (			A__	:   Dict     )    ->      str:





      """simple docstring"""



      _lowercase	       ={}

      _lowercase	       =job['started_at']
      _lowercase	       =job['completed_at']

      _lowercase	       =date_parser.parse(A__     )
      _lowercase	       =date_parser.parse(A__     )

      _lowercase	       =round((end_datetime - start_datetime).total_seconds() / 60.0     )

      _lowercase	       =start
      _lowercase	       =end
      _lowercase	       =duration_in_min

      return job_info
def      a      (			A__	:   Dict      ,   A__	:   str=None     )    ->      Tuple:





      """simple docstring"""



      _lowercase	       =None
      if token is not None:
            _lowercase	       ={'Accept': 'application/vnd.github+json', 'Authorization': F'''Bearer {token}'''}

      _lowercase	       =F'''https://api.github.com/repos/huggingface/transformers/actions/runs/{workflow_run_id}/jobs?per_page=100'''
      _lowercase	       =requests.get(A__      ,   headers=A__     ).json()
      _lowercase	       ={}

      try:
            job_time.update({job['name']: extract_time_from_single_job(A__     ) for job in result['jobs']}     )
            _lowercase	       =math.ceil((result['total_count'] - 100) / 100     )

            for i in range(A__     ):
                  _lowercase	       =requests.get(url + F'''&page={i + 2}'''      ,   headers=A__     ).json()
                  job_time.update({job['name']: extract_time_from_single_job(A__     ) for job in result['jobs']}     )

            return job_time
      except Exception:
            print(F'''Unknown error, could not fetch links:\n{traceback.format_exc()}'''     )

      return {}


if __name__ == "__main__":
  lowercase_									=   argparse.ArgumentParser()
  # Required parameters
  parser.add_argument('--workflow_run_id', type=str, required=True, help='A GitHub Actions workflow run id.')
  lowercase_									=   parser.parse_args()

  lowercase_									=   get_job_time(args.workflow_run_id)
  lowercase_									=   dict(sorted(job_time.items(), key=lambda item: item[1]["duration"], reverse=True))

  for k, v in job_time.items():
    print(f"{k}: {v['duration']}")


