





from __future__ import annotations
def 					_A	(		SCREAMING_SNAKE_CASE					:       int	,	SCREAMING_SNAKE_CASE					:       int						):





  """simple docstring"""



  if partitions <= 0:
    raise ValueError("partitions must be a positive number!"						)
  if partitions > number_of_bytes:
    raise ValueError("partitions can not > number_of_bytes!"						)
  a__     : str  =number_of_bytes // partitions
  a__     : Any  =[]
  for i in range(SCREAMING_SNAKE_CASE						):
    a__     : List[Any]  =i * bytes_per_partition + 1
    a__     : str  =(
        number_of_bytes if i == partitions - 1 else (i + 1) * bytes_per_partition
    )
    allocation_list.append(f'''{start_bytes}-{end_bytes}'''						)
  return allocation_list


if __name__ == "__main__":
     import doctest

     doctest.testmod()













import inspect
import unittest

from transformers import MobileNetVaConfig
from transformers.testing_utils import require_torch, require_vision, slow, torch_device
from transformers.utils import cached_property, is_torch_available, is_vision_available

from ...test_configuration_common import ConfigTester
from ...test_modeling_common import ModelTesterMixin, floats_tensor, ids_tensor
from ...test_pipeline_mixin import PipelineTesterMixin


if is_torch_available():
     import torch

     from transformers import MobileNetVaForImageClassification, MobileNetVaModel
     from transformers.models.mobilenet_va.modeling_mobilenet_va import MOBILENET_V1_PRETRAINED_MODEL_ARCHIVE_LIST


if is_vision_available():
     from PIL import Image

     from transformers import MobileNetVaImageProcessor







class     __lowerCAmelCase	(   UpperCamelCase__):







      def        _lowercase  (   self							)       ->						Any:



        '''simple docstring'''




        a__     : Optional[int]  =self.config_class(**self.inputs_dict							)
        self.parent.assertTrue(hasattr(lowerCAmelCase__		,				"tf_padding"							)							)
        self.parent.assertTrue(hasattr(lowerCAmelCase__		,				"depth_multiplier"							)							)







class     __lowerCAmelCase	:







      def __init__(   self		,				lowerCAmelCase__		,				lowerCAmelCase__=1_3		,				lowerCAmelCase__=3		,				lowerCAmelCase__=3_2		,				lowerCAmelCase__=0.25		,				lowerCAmelCase__=8		,				lowerCAmelCase__=True		,				lowerCAmelCase__=1_0_2_4		,				lowerCAmelCase__=3_2		,				lowerCAmelCase__="relu6"		,				lowerCAmelCase__=0.1		,				lowerCAmelCase__=0.02		,				lowerCAmelCase__=True		,				lowerCAmelCase__=True		,				lowerCAmelCase__=1_0		,				lowerCAmelCase__=None		,				)       ->						Dict:



        '''simple docstring'''




        a__     : int  =parent
        a__     : Optional[Any]  =batch_size
        a__     : Tuple  =num_channels
        a__     : Dict  =image_size
        a__     : Union[str, Any]  =depth_multiplier
        a__     : List[str]  =min_depth
        a__     : Dict  =tf_padding
        a__     : Any  =int(last_hidden_size * depth_multiplier							)
        a__     : Tuple  =output_stride
        a__     : Optional[Any]  =hidden_act
        a__     : str  =classifier_dropout_prob
        a__     : int  =use_labels
        a__     : List[Any]  =is_training
        a__     : List[str]  =num_labels
        a__     : Dict  =initializer_range
        a__     : Tuple  =scope







      def        _lowercase  (   self							)       ->						List[str]:



        '''simple docstring'''




        a__     : Union[str, Any]  =floats_tensor([self.batch_size, self.num_channels, self.image_size, self.image_size]							)

        a__     : Any  =None
        a__     : List[Any]  =None
        if self.use_labels:
          a__     : Dict  =ids_tensor([self.batch_size]		,				self.num_labels							)
          a__     : Tuple  =ids_tensor([self.batch_size, self.image_size, self.image_size]		,				self.num_labels							)

        a__     : Optional[Any]  =self.get_config()

        return config, pixel_values, labels, pixel_labels







      def        _lowercase  (   self							)       ->						str:



        '''simple docstring'''




        return MobileNetVaConfig(
            num_channels=self.num_channels		,				image_size=self.image_size		,				depth_multiplier=self.depth_multiplier		,				min_depth=self.min_depth		,				tf_padding=self.tf_padding		,				hidden_act=self.hidden_act		,				classifier_dropout_prob=self.classifier_dropout_prob		,				initializer_range=self.initializer_range		,				)







      def        _lowercase  (   self		,				lowerCAmelCase__		,				lowerCAmelCase__		,				lowerCAmelCase__		,				lowerCAmelCase__							)       ->						Union[str, Any]:



        '''simple docstring'''




        a__     : Union[str, Any]  =MobileNetVaModel(config=lowerCAmelCase__							)
        model.to(lowerCAmelCase__							)
        model.eval()
        a__     : str  =model(lowerCAmelCase__							)
        self.parent.assertEqual(
            result.last_hidden_state.shape		,				(
                self.batch_size,
                self.last_hidden_size,
                self.image_size // self.output_stride,
                self.image_size // self.output_stride,
            )		,				)







      def        _lowercase  (   self		,				lowerCAmelCase__		,				lowerCAmelCase__		,				lowerCAmelCase__		,				lowerCAmelCase__							)       ->						str:



        '''simple docstring'''




        a__     : Any  =self.num_labels
        a__     : Dict  =MobileNetVaForImageClassification(lowerCAmelCase__							)
        model.to(lowerCAmelCase__							)
        model.eval()
        a__     : Optional[int]  =model(lowerCAmelCase__		,				labels=lowerCAmelCase__							)
        self.parent.assertEqual(result.logits.shape		,				(self.batch_size, self.num_labels)							)







      def        _lowercase  (   self							)       ->						str:



        '''simple docstring'''




        a__     : str  =self.prepare_config_and_inputs()
        a__ ,							a__ ,							a__ ,							a__     : List[str]  =config_and_inputs
        a__     : Tuple  ={"pixel_values": pixel_values}
        return config, inputs_dict







@require_torch
class     __lowerCAmelCase	(   UpperCamelCase__						,    UpperCamelCase__						,    unittest.TestCase):
      _lowercase    : List[str]            =      (MobileNetVaModel, MobileNetVaForImageClassification) if is_torch_available() else ()
      _lowercase    : Optional[int]            =      (
          {"""feature-extraction""": MobileNetVaModel, """image-classification""": MobileNetVaForImageClassification}
          if is_torch_available()
          else {}
      )

      _lowercase    : Union[str, Any]            =      False
      _lowercase    : Optional[int]            =      False
      _lowercase    : List[str]            =      False
      _lowercase    : str            =      False







      def        _lowercase  (   self							)       ->						Optional[int]:



        '''simple docstring'''




        a__     : Optional[Any]  =MobileNetVaModelTester(self							)
        a__     : Optional[Any]  =MobileNetVaConfigTester(self		,				config_class=lowerCAmelCase__		,				has_text_modality=lowerCAmelCase__							)







      def        _lowercase  (   self							)       ->						Dict:



        '''simple docstring'''




        self.config_tester.run_common_tests()







      @unittest.skip(reason="MobileNetV1 does not use inputs_embeds"							)
      def        _lowercase  (   self							)       ->						Union[str, Any]:



        '''simple docstring'''




        pass







      @unittest.skip(reason="MobileNetV1 does not support input and output embeddings"							)
      def        _lowercase  (   self							)       ->						List[str]:



        '''simple docstring'''




        pass







      @unittest.skip(reason="MobileNetV1 does not output attentions"							)
      def        _lowercase  (   self							)       ->						Any:



        '''simple docstring'''




        pass







      def        _lowercase  (   self							)       ->						str:



        '''simple docstring'''




        a__ ,							a__     : Optional[int]  =self.model_tester.prepare_config_and_inputs_for_common()

        for model_class in self.all_model_classes:
          a__     : Optional[int]  =model_class(lowerCAmelCase__							)
          a__     : str  =inspect.signature(model.forward							)
          # signature.parameters is an OrderedDict => so arg_names order is deterministic
          a__     : Optional[int]  =[*signature.parameters.keys()]

          a__     : Union[str, Any]  =["pixel_values"]
          self.assertListEqual(arg_names[:1]		,				lowerCAmelCase__							)







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



        '''simple docstring'''




        a__     : Optional[Any]  =self.model_tester.prepare_config_and_inputs()
        self.model_tester.create_and_check_model(*lowerCAmelCase__							)







      def        _lowercase  (   self							)       ->						List[str]:



        '''simple docstring'''




        def check_hidden_states_output(lowerCAmelCase__		,				lowerCAmelCase__		,				lowerCAmelCase__							):
          a__     : int  =model_class(lowerCAmelCase__							)
          model.to(lowerCAmelCase__							)
          model.eval()

          with torch.no_grad():
            a__     : List[Any]  =model(**self._prepare_for_class(lowerCAmelCase__		,				lowerCAmelCase__							)							)

          a__     : Optional[int]  =outputs.hidden_states

          a__     : Optional[int]  =2_6
          self.assertEqual(len(lowerCAmelCase__							)		,				lowerCAmelCase__							)

        a__ ,							a__     : Optional[Any]  =self.model_tester.prepare_config_and_inputs_for_common()

        for model_class in self.all_model_classes:
          a__     : Optional[int]  =True
          check_hidden_states_output(lowerCAmelCase__		,				lowerCAmelCase__		,				lowerCAmelCase__							)

          # check that output_hidden_states also work using config
          del inputs_dict["output_hidden_states"]
          a__     : Union[str, Any]  =True

          check_hidden_states_output(lowerCAmelCase__		,				lowerCAmelCase__		,				lowerCAmelCase__							)







      def        _lowercase  (   self							)       ->						Any:



        '''simple docstring'''




        a__     : Dict  =self.model_tester.prepare_config_and_inputs()
        self.model_tester.create_and_check_for_image_classification(*lowerCAmelCase__							)







      @slow
      def        _lowercase  (   self							)       ->						Any:



        '''simple docstring'''




        for model_name in MOBILENET_V1_PRETRAINED_MODEL_ARCHIVE_LIST[:1]:
          a__     : List[str]  =MobileNetVaModel.from_pretrained(lowerCAmelCase__							)
          self.assertIsNotNone(lowerCAmelCase__							)
def 					_A	(		):





  """simple docstring"""



  a__     : Optional[Any]  =Image.open("./tests/fixtures/tests_samples/COCO/000000039769.png"						)
  return image







@require_torch
@require_vision
class     __lowerCAmelCase	(   unittest.TestCase):







      @cached_property
      def        _lowercase  (   self							)       ->						Optional[Any]:



        '''simple docstring'''




        return (
            MobileNetVaImageProcessor.from_pretrained("google/mobilenet_v1_1.0_224"							) if is_vision_available() else None
        )







      @slow
      def        _lowercase  (   self							)       ->						Optional[int]:



        '''simple docstring'''




        a__     : str  =MobileNetVaForImageClassification.from_pretrained("google/mobilenet_v1_1.0_224"							).to(lowerCAmelCase__							)

        a__     : Optional[Any]  =self.default_image_processor
        a__     : Optional[int]  =prepare_img()
        a__     : Optional[int]  =image_processor(images=lowerCAmelCase__		,				return_tensors="pt"							).to(lowerCAmelCase__							)

        # forward pass
        with torch.no_grad():
          a__     : int  =model(**lowerCAmelCase__							)

        # verify the logits
        a__     : str  =torch.Size((1, 1_0_0_1)							)
        self.assertEqual(outputs.logits.shape		,				lowerCAmelCase__							)

        a__     : int  =torch.tensor([-4.17_39, -1.12_33, 3.12_05]							).to(lowerCAmelCase__							)

        self.assertTrue(torch.allclose(outputs.logits[0, :3]		,				lowerCAmelCase__		,				atol=1E-4							)							)






