





'''simple docstring'''

import os

import pytest
from attr import dataclass


lowercase__						:      Optional[Any]	=    'us-east-1'  # defaults region
@dataclass
class     __lowerCAmelCase					:






							"""simple docstring"""
							_snake_case     :					str
							_snake_case     :					Any						    =   'arn:aws:iam::558105141721:role/sagemaker_execution_role'
							_snake_case     :					Any						    =   {
							    'task_name': 'mnli',
							    'per_device_train_batch_size': 1_6,
							    'per_device_eval_batch_size': 1_6,
							    'do_train': True,
							    'do_eval': True,
							    'do_predict': True,
							    'output_dir': '/opt/ml/model',
							    'overwrite_output_dir': True,
							    'max_steps': 5_0_0,
							    'save_steps': 5_5_0_0,
							}
							_snake_case     :					Union[str, Any]						    =   {**hyperparameters, 'max_steps': 1_0_0_0}





							@property
							def snake_case__						(							self       :		str					)   ->					str:

												'''simple docstring'''




												if self.framework == "pytorch":
																	return [
																	    {"Name": "train_runtime", "Regex": r"train_runtime.*=\D*(.*?)$"},
																	    {"Name": "eval_accuracy", "Regex": r"eval_accuracy.*=\D*(.*?)$"},
																	    {"Name": "eval_loss", "Regex": r"eval_loss.*=\D*(.*?)$"},
																	]
												else:
																	return [
																	    {"Name": "train_runtime", "Regex": r"train_runtime.*=\D*(.*?)$"},
																	    {"Name": "eval_accuracy", "Regex": r"loss.*=\D*(.*?)]?$"},
																	    {"Name": "eval_loss", "Regex": r"sparse_categorical_accuracy.*=\D*(.*?)]?$"},
																	]





							@property
							def snake_case__						(							self       :		Union[str, Any]					)   ->					str:

												'''simple docstring'''




												return f"""{self.framework}-transfromers-test"""





							@property
							def snake_case__						(							self       :		List[Any]					)   ->					str:

												'''simple docstring'''




												return f"""./tests/sagemaker/scripts/{self.framework}"""







							@property
							def snake_case__						(							self       :		Optional[Any]					)   ->					str:

												'''simple docstring'''




												if self.framework == "pytorch":
																	return "763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-pytorch-training:1.7.1-transformers4.6.1-gpu-py36-cu110-ubuntu18.04"
												else:
																	return "763104351884.dkr.ecr.us-east-1.amazonaws.com/huggingface-tensorflow-training:2.4.1-transformers4.6.1-gpu-py37-cu110-ubuntu18.04"







@pytest.fixture(scope='''class'''				)
def  a__    (    lowercase :					int				)		->		Tuple:





					"""simple docstring"""

					_UpperCamelCase       =	SageMakerTestEnvironment(framework=request.cls.framework				)










'''simple docstring'''

import unittest

from diffusers.pipelines.pipeline_utils import is_safetensors_compatible
class     __lowerCAmelCase					(   unittest.TestCase			):






       """simple docstring"""





       def snake_case__						(							self       :		Tuple					)   ->					int:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''safety_checker/pytorch_model.bin''',
                '''safety_checker/model.safetensors''',
                '''vae/diffusion_pytorch_model.bin''',
                '''vae/diffusion_pytorch_model.safetensors''',
                '''text_encoder/pytorch_model.bin''',
                '''text_encoder/model.safetensors''',
                '''unet/diffusion_pytorch_model.bin''',
                '''unet/diffusion_pytorch_model.safetensors''',
            ]
            self.assertTrue(is_safetensors_compatible(lowerCAmelCase__					)					)





       def snake_case__						(							self       :		int					)   ->					Tuple:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''unet/diffusion_pytorch_model.bin''',
                '''unet/diffusion_pytorch_model.safetensors''',
            ]
            self.assertTrue(is_safetensors_compatible(lowerCAmelCase__					)					)





       def snake_case__						(							self       :		Optional[Any]					)   ->					List[str]:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''safety_checker/pytorch_model.bin''',
                '''safety_checker/model.safetensors''',
                '''vae/diffusion_pytorch_model.bin''',
                '''vae/diffusion_pytorch_model.safetensors''',
                '''text_encoder/pytorch_model.bin''',
                '''text_encoder/model.safetensors''',
                '''unet/diffusion_pytorch_model.bin''',
                # Removed: 'unet/diffusion_pytorch_model.safetensors',
            ]
            self.assertFalse(is_safetensors_compatible(lowerCAmelCase__					)					)





       def snake_case__						(							self       :		List[Any]					)   ->					List[Any]:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''text_encoder/pytorch_model.bin''',
                '''text_encoder/model.safetensors''',
            ]
            self.assertTrue(is_safetensors_compatible(lowerCAmelCase__					)					)





       def snake_case__						(							self       :		Dict					)   ->					Dict:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''safety_checker/pytorch_model.bin''',
                '''safety_checker/model.safetensors''',
                '''vae/diffusion_pytorch_model.bin''',
                '''vae/diffusion_pytorch_model.safetensors''',
                '''text_encoder/pytorch_model.bin''',
                # Removed: 'text_encoder/model.safetensors',
                '''unet/diffusion_pytorch_model.bin''',
                '''unet/diffusion_pytorch_model.safetensors''',
            ]
            self.assertFalse(is_safetensors_compatible(lowerCAmelCase__					)					)





       def snake_case__						(							self       :		Any					)   ->					Optional[int]:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''safety_checker/pytorch_model.fp16.bin''',
                '''safety_checker/model.fp16.safetensors''',
                '''vae/diffusion_pytorch_model.fp16.bin''',
                '''vae/diffusion_pytorch_model.fp16.safetensors''',
                '''text_encoder/pytorch_model.fp16.bin''',
                '''text_encoder/model.fp16.safetensors''',
                '''unet/diffusion_pytorch_model.fp16.bin''',
                '''unet/diffusion_pytorch_model.fp16.safetensors''',
            ]
            _UpperCamelCase       =	'''fp16'''
            self.assertTrue(is_safetensors_compatible(lowerCAmelCase__	,			variant=lowerCAmelCase__					)					)





       def snake_case__						(							self       :		Optional[Any]					)   ->					Dict:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''unet/diffusion_pytorch_model.fp16.bin''',
                '''unet/diffusion_pytorch_model.fp16.safetensors''',
            ]
            _UpperCamelCase       =	'''fp16'''
            self.assertTrue(is_safetensors_compatible(lowerCAmelCase__	,			variant=lowerCAmelCase__					)					)





       def snake_case__						(							self       :		Tuple					)   ->					Optional[Any]:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''unet/diffusion_pytorch_model.bin''',
                '''unet/diffusion_pytorch_model.safetensors''',
            ]
            _UpperCamelCase       =	'''fp16'''
            self.assertTrue(is_safetensors_compatible(lowerCAmelCase__	,			variant=lowerCAmelCase__					)					)





       def snake_case__						(							self       :		Tuple					)   ->					Union[str, Any]:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''safety_checker/pytorch_model.fp16.bin''',
                '''safety_checker/model.fp16.safetensors''',
                '''vae/diffusion_pytorch_model.fp16.bin''',
                '''vae/diffusion_pytorch_model.fp16.safetensors''',
                '''text_encoder/pytorch_model.fp16.bin''',
                '''text_encoder/model.fp16.safetensors''',
                '''unet/diffusion_pytorch_model.fp16.bin''',
                # Removed: 'unet/diffusion_pytorch_model.fp16.safetensors',
            ]
            _UpperCamelCase       =	'''fp16'''
            self.assertFalse(is_safetensors_compatible(lowerCAmelCase__	,			variant=lowerCAmelCase__					)					)





       def snake_case__						(							self       :		Optional[int]					)   ->					Dict:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''text_encoder/pytorch_model.fp16.bin''',
                '''text_encoder/model.fp16.safetensors''',
            ]
            _UpperCamelCase       =	'''fp16'''
            self.assertTrue(is_safetensors_compatible(lowerCAmelCase__	,			variant=lowerCAmelCase__					)					)





       def snake_case__						(							self       :		Optional[Any]					)   ->					str:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''text_encoder/pytorch_model.bin''',
                '''text_encoder/model.safetensors''',
            ]
            _UpperCamelCase       =	'''fp16'''
            self.assertTrue(is_safetensors_compatible(lowerCAmelCase__	,			variant=lowerCAmelCase__					)					)







       def snake_case__						(							self       :		List[str]					)   ->					Optional[int]:

            '''simple docstring'''




            _UpperCamelCase       =	[
                '''safety_checker/pytorch_model.fp16.bin''',
                '''safety_checker/model.fp16.safetensors''',
                '''vae/diffusion_pytorch_model.fp16.bin''',
                '''vae/diffusion_pytorch_model.fp16.safetensors''',
                '''text_encoder/pytorch_model.fp16.bin''',
                # 'text_encoder/model.fp16.safetensors',
                '''unet/diffusion_pytorch_model.fp16.bin''',
                '''unet/diffusion_pytorch_model.fp16.safetensors''',
            ]
            _UpperCamelCase       =	'''fp16'''
            self.assertFalse(is_safetensors_compatible(lowerCAmelCase__	,			variant=lowerCAmelCase__					)					)



