



from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


lowerCamelCase__			= {
    """configuration_blip_2""": [
        """BLIP_2_PRETRAINED_CONFIG_ARCHIVE_MAP""",
        """Blip2Config""",
        """Blip2QFormerConfig""",
        """Blip2VisionConfig""",
    ],
    """processing_blip_2""": ["""Blip2Processor"""],
}

try:
      if not is_torch_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      lowerCamelCase__			= [
          """BLIP_2_PRETRAINED_MODEL_ARCHIVE_LIST""",
          """Blip2Model""",
          """Blip2QFormerModel""",
          """Blip2PreTrainedModel""",
          """Blip2ForConditionalGeneration""",
          """Blip2VisionModel""",
      ]

if TYPE_CHECKING:
      from .configuration_blip_a import (
          BLIP_2_PRETRAINED_CONFIG_ARCHIVE_MAP,
          BlipaConfig,
          BlipaQFormerConfig,
          BlipaVisionConfig,
      )
      from .processing_blip_a import BlipaProcessor

      try:
            if not is_torch_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_blip_a import (
                BLIP_2_PRETRAINED_MODEL_ARCHIVE_LIST,
                BlipaForConditionalGeneration,
                BlipaModel,
                BlipaPreTrainedModel,
                BlipaQFormerModel,
                BlipaVisionModel,
            )

else:
      import sys

      lowerCamelCase__			= _LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)





import unittest
from dataclasses import dataclass

import pytest

from accelerate.commands.config.config_args import SageMakerConfig
from accelerate.utils import ComputeEnvironment
from accelerate.utils.launch import _convert_nargs_to_dict






@dataclass
class    SCREAMING_SNAKE_CASE			(							lowerCamelCase__					):
							__lowerCamelCase	:	List[Any]    =ComputeEnvironment.AMAZON_SAGEMAKER
							__lowerCamelCase	:	Optional[int]    =True
							__lowerCamelCase	:	List[Any]    ='ml.p3.2xlarge'
							__lowerCamelCase	:	Optional[int]    ='accelerate_sagemaker_execution_role'
							__lowerCamelCase	:	Optional[Any]    ='hf-sm'
							__lowerCamelCase	:	Any    ='us-east-1'
							__lowerCamelCase	:	Any    =1
							__lowerCamelCase	:	List[str]    ='accelerate-sagemaker-1'
							__lowerCamelCase	:	Union[str, Any]    ='1.6'
							__lowerCamelCase	:	str    ='4.4'
							__lowerCamelCase	:	List[str]    ='train.py'
							__lowerCamelCase	:	List[Any]    =[
							    '--model_name_or_path',
							    'bert',
							    '--do_train',
							    'False',
							    '--epochs',
							    '3',
							    '--learning_rate',
							    '5e-5',
							    '--max_steps',
							    '50.5',
							]
							__lowerCamelCase	:	Tuple    =[
							    '--model_name_or_path',
							    'bert',
							    '--do_train',
							    '--do_test',
							    'False',
							    '--do_predict',
							    '--epochs',
							    '3',
							    '--learning_rate',
							    '5e-5',
							    '--max_steps',
							    '50.5',
							]






class    SCREAMING_SNAKE_CASE			(							unittest.TestCase					):





							def 						UpperCamelCase_					(   self		:		Union[str, Any]	):

											'''simple docstring'''
											# If no defaults are changed, `to_kwargs` returns an empty dict.
											__a       =		_convert_nargs_to_dict(MockLaunchConfig.success_training_script_args	)
											assert isinstance(converted_args["""model_name_or_path"""]      ,				__lowercase	)
											assert isinstance(converted_args["""do_train"""]      ,				__lowercase	)
											assert isinstance(converted_args["""epochs"""]      ,				__lowercase	)
											assert isinstance(converted_args["""learning_rate"""]      ,				__lowercase	)
											assert isinstance(converted_args["""max_steps"""]      ,				__lowercase	)

											with pytest.raises(__lowercase	):
															_convert_nargs_to_dict(MockLaunchConfig.fail_training_script_args	)
