
import unittest

from transformers import MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPING, is_vision_available, pipeline
from transformers.testing_utils import (
    is_pipeline_test,
    nested_simplify,
    require_tf,
    require_torch,
    require_vision,
    slow,
)

from .test_pipelines_common import ANY


if is_vision_available():
      from PIL import Image
else:

      class __lowerCAmelCase	:





           @staticmethod
           def 		UpperCAmelCase      (			*__UpperCAmelCase			,  **__UpperCAmelCase   ):

            '''simple docstring'''




            pass

@is_pipeline_test
@require_vision
@require_torch
class __lowerCAmelCase	(   unittest.TestCase    ):
     lowercase		     =		MODEL_FOR_ZERO_SHOT_OBJECT_DETECTION_MAPPING





     def 		UpperCAmelCase      (			self			,  __UpperCAmelCase			,  __UpperCAmelCase			,  __UpperCAmelCase   ):

      '''simple docstring'''




      __UpperCamelCase       						= pipeline(
          'zero-shot-object-detection'			,  model='hf-internal-testing/tiny-random-owlvit-object-detection'   )

      __UpperCamelCase       						= [
          {
              'image': './tests/fixtures/tests_samples/COCO/000000039769.png',
              'candidate_labels': ['cat', 'remote', 'couch'],
          }
      ]
      return object_detector, examples





     def 		UpperCAmelCase      (			self			,  __UpperCAmelCase			,  __UpperCAmelCase   ):

      '''simple docstring'''




      __UpperCamelCase       						= object_detector(examples[0]			,  threshold=0.0   )

      __UpperCamelCase       						= len(__lowerCAmelCase   )
      self.assertGreater(__lowerCAmelCase			,  0   )
      self.assertEqual(
          __lowerCAmelCase			,  [
              {
                  'score': ANY(__lowerCAmelCase   ),
                  'label': ANY(__lowerCAmelCase   ),
                  'box': {'xmin': ANY(__lowerCAmelCase   ), 'ymin': ANY(__lowerCAmelCase   ), 'xmax': ANY(__lowerCAmelCase   ), 'ymax': ANY(__lowerCAmelCase   )},
              }
              for i in range(__lowerCAmelCase   )
          ]			,  )





     @require_tf
     @unittest.skip('Zero Shot Object Detection not implemented in TF'   )
     def 		UpperCAmelCase      (			self   ):

      '''simple docstring'''




      pass





     @require_torch
     def 		UpperCAmelCase      (			self   ):

      '''simple docstring'''




      __UpperCamelCase       						= pipeline(
          'zero-shot-object-detection'			,  model='hf-internal-testing/tiny-random-owlvit-object-detection'   )

      __UpperCamelCase       						= object_detector(
          './tests/fixtures/tests_samples/COCO/000000039769.png'			,  candidate_labels=['cat', 'remote', 'couch']			,  threshold=0.6_4			,  )

      self.assertEqual(
          nested_simplify(__lowerCAmelCase			,  decimals=4   )			,  [
              {'score': 0.7_2_3_5, 'label': 'cat', 'box': {'xmin': 204, 'ymin': 167, 'xmax': 232, 'ymax': 190}},
              {'score': 0.7_2_1_8, 'label': 'remote', 'box': {'xmin': 204, 'ymin': 167, 'xmax': 232, 'ymax': 190}},
              {'score': 0.7_1_8_4, 'label': 'couch', 'box': {'xmin': 204, 'ymin': 167, 'xmax': 232, 'ymax': 190}},
              {'score': 0.6_7_4_8, 'label': 'remote', 'box': {'xmin': 571, 'ymin': 83, 'xmax': 598, 'ymax': 103}},
              {'score': 0.6_6_5_6, 'label': 'cat', 'box': {'xmin': 571, 'ymin': 83, 'xmax': 598, 'ymax': 103}},
              {'score': 0.6_6_1_4, 'label': 'couch', 'box': {'xmin': 571, 'ymin': 83, 'xmax': 598, 'ymax': 103}},
              {'score': 0.6_4_5_6, 'label': 'remote', 'box': {'xmin': 494, 'ymin': 105, 'xmax': 521, 'ymax': 127}},
              {'score': 0.6_4_2, 'label': 'remote', 'box': {'xmin': 67, 'ymin': 274, 'xmax': 93, 'ymax': 297}},
              {'score': 0.6_4_1_9, 'label': 'cat', 'box': {'xmin': 494, 'ymin': 105, 'xmax': 521, 'ymax': 127}},
          ]			,  )

      __UpperCamelCase       						= object_detector(
          [
              {
                  'image': './tests/fixtures/tests_samples/COCO/000000039769.png',
                  'candidate_labels': ['cat', 'remote', 'couch'],
              }
          ]			,  threshold=0.6_4			,  )

      self.assertEqual(
          nested_simplify(__lowerCAmelCase			,  decimals=4   )			,  [
              [
                  {'score': 0.7_2_3_5, 'label': 'cat', 'box': {'xmin': 204, 'ymin': 167, 'xmax': 232, 'ymax': 190}},
                  {'score': 0.7_2_1_8, 'label': 'remote', 'box': {'xmin': 204, 'ymin': 167, 'xmax': 232, 'ymax': 190}},
                  {'score': 0.7_1_8_4, 'label': 'couch', 'box': {'xmin': 204, 'ymin': 167, 'xmax': 232, 'ymax': 190}},
                  {'score': 0.6_7_4_8, 'label': 'remote', 'box': {'xmin': 571, 'ymin': 83, 'xmax': 598, 'ymax': 103}},
                  {'score': 0.6_6_5_6, 'label': 'cat', 'box': {'xmin': 571, 'ymin': 83, 'xmax': 598, 'ymax': 103}},
                  {'score': 0.6_6_1_4, 'label': 'couch', 'box': {'xmin': 571, 'ymin': 83, 'xmax': 598, 'ymax': 103}},
                  {'score': 0.6_4_5_6, 'label': 'remote', 'box': {'xmin': 494, 'ymin': 105, 'xmax': 521, 'ymax': 127}},
                  {'score': 0.6_4_2, 'label': 'remote', 'box': {'xmin': 67, 'ymin': 274, 'xmax': 93, 'ymax': 297}},
                  {'score': 0.6_4_1_9, 'label': 'cat', 'box': {'xmin': 494, 'ymin': 105, 'xmax': 521, 'ymax': 127}},
              ]
          ]			,  )





     @require_torch
     @slow
     def 		UpperCAmelCase      (			self   ):

      '''simple docstring'''




      __UpperCamelCase       						= pipeline('zero-shot-object-detection'   )

      __UpperCamelCase       						= object_detector(
          'http://images.cocodataset.org/val2017/000000039769.jpg'			,  candidate_labels=['cat', 'remote', 'couch']			,  )
      self.assertEqual(
          nested_simplify(__lowerCAmelCase			,  decimals=4   )			,  [
              {'score': 0.2_8_6_8, 'label': 'cat', 'box': {'xmin': 324, 'ymin': 20, 'xmax': 640, 'ymax': 373}},
              {'score': 0.2_7_7, 'label': 'remote', 'box': {'xmin': 40, 'ymin': 72, 'xmax': 177, 'ymax': 115}},
              {'score': 0.2_5_3_7, 'label': 'cat', 'box': {'xmin': 1, 'ymin': 55, 'xmax': 315, 'ymax': 472}},
              {'score': 0.1_4_7_4, 'label': 'remote', 'box': {'xmin': 335, 'ymin': 74, 'xmax': 371, 'ymax': 187}},
              {'score': 0.1_2_0_8, 'label': 'couch', 'box': {'xmin': 4, 'ymin': 0, 'xmax': 642, 'ymax': 476}},
          ]			,  )

      __UpperCamelCase       						= object_detector(
          [
              {
                  'image': 'http://images.cocodataset.org/val2017/000000039769.jpg',
                  'candidate_labels': ['cat', 'remote', 'couch'],
              },
              {
                  'image': 'http://images.cocodataset.org/val2017/000000039769.jpg',
                  'candidate_labels': ['cat', 'remote', 'couch'],
              },
          ]			,  )
      self.assertEqual(
          nested_simplify(__lowerCAmelCase			,  decimals=4   )			,  [
              [
                  {'score': 0.2_8_6_8, 'label': 'cat', 'box': {'xmin': 324, 'ymin': 20, 'xmax': 640, 'ymax': 373}},
                  {'score': 0.2_7_7, 'label': 'remote', 'box': {'xmin': 40, 'ymin': 72, 'xmax': 177, 'ymax': 115}},
                  {'score': 0.2_5_3_7, 'label': 'cat', 'box': {'xmin': 1, 'ymin': 55, 'xmax': 315, 'ymax': 472}},
                  {'score': 0.1_4_7_4, 'label': 'remote', 'box': {'xmin': 335, 'ymin': 74, 'xmax': 371, 'ymax': 187}},
                  {'score': 0.1_2_0_8, 'label': 'couch', 'box': {'xmin': 4, 'ymin': 0, 'xmax': 642, 'ymax': 476}},
              ],
              [
                  {'score': 0.2_8_6_8, 'label': 'cat', 'box': {'xmin': 324, 'ymin': 20, 'xmax': 640, 'ymax': 373}},
                  {'score': 0.2_7_7, 'label': 'remote', 'box': {'xmin': 40, 'ymin': 72, 'xmax': 177, 'ymax': 115}},
                  {'score': 0.2_5_3_7, 'label': 'cat', 'box': {'xmin': 1, 'ymin': 55, 'xmax': 315, 'ymax': 472}},
                  {'score': 0.1_4_7_4, 'label': 'remote', 'box': {'xmin': 335, 'ymin': 74, 'xmax': 371, 'ymax': 187}},
                  {'score': 0.1_2_0_8, 'label': 'couch', 'box': {'xmin': 4, 'ymin': 0, 'xmax': 642, 'ymax': 476}},
              ],
          ]			,  )





     @require_tf
     @unittest.skip('Zero Shot Object Detection not implemented in TF'   )
     def 		UpperCAmelCase      (			self   ):

      '''simple docstring'''




      pass





     @require_torch
     @slow
     def 		UpperCAmelCase      (			self   ):

      '''simple docstring'''




      __UpperCamelCase       						= 0.2
      __UpperCamelCase       						= pipeline('zero-shot-object-detection'   )

      __UpperCamelCase       						= object_detector(
          'http://images.cocodataset.org/val2017/000000039769.jpg'			,  candidate_labels=['cat', 'remote', 'couch']			,  threshold=__lowerCAmelCase			,  )
      self.assertEqual(
          nested_simplify(__lowerCAmelCase			,  decimals=4   )			,  [
              {'score': 0.2_8_6_8, 'label': 'cat', 'box': {'xmin': 324, 'ymin': 20, 'xmax': 640, 'ymax': 373}},
              {'score': 0.2_7_7, 'label': 'remote', 'box': {'xmin': 40, 'ymin': 72, 'xmax': 177, 'ymax': 115}},
              {'score': 0.2_5_3_7, 'label': 'cat', 'box': {'xmin': 1, 'ymin': 55, 'xmax': 315, 'ymax': 472}},
          ]			,  )







     @require_torch
     @slow
     def 		UpperCAmelCase      (			self   ):

      '''simple docstring'''




      __UpperCamelCase       						= 2
      __UpperCamelCase       						= pipeline('zero-shot-object-detection'   )

      __UpperCamelCase       						= object_detector(
          'http://images.cocodataset.org/val2017/000000039769.jpg'			,  candidate_labels=['cat', 'remote', 'couch']			,  top_k=__lowerCAmelCase			,  )
      self.assertEqual(
          nested_simplify(__lowerCAmelCase			,  decimals=4   )			,  [
              {'score': 0.2_8_6_8, 'label': 'cat', 'box': {'xmin': 324, 'ymin': 20, 'xmax': 640, 'ymax': 373}},
              {'score': 0.2_7_7, 'label': 'remote', 'box': {'xmin': 40, 'ymin': 72, 'xmax': 177, 'ymax': 115}},
          ]			,  )





"""simple docstring"""

from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_flax_available,
    is_sentencepiece_available,
    is_tf_available,
    is_tokenizers_available,
    is_torch_available,
)


UpperCamelCase			:       List[Any]									=				{
    "configuration_xlm_roberta": [
        "XLM_ROBERTA_PRETRAINED_CONFIG_ARCHIVE_MAP",
        "XLMRobertaConfig",
        "XLMRobertaOnnxConfig",
    ],
}

try:
      if not is_sentencepiece_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase			:       Optional[Any]									=				["XLMRobertaTokenizer"]

try:
      if not is_tokenizers_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase			:       int									=				["XLMRobertaTokenizerFast"]

try:
      if not is_torch_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase			:       Optional[int]									=				[
          "XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST",
          "XLMRobertaForCausalLM",
          "XLMRobertaForMaskedLM",
          "XLMRobertaForMultipleChoice",
          "XLMRobertaForQuestionAnswering",
          "XLMRobertaForSequenceClassification",
          "XLMRobertaForTokenClassification",
          "XLMRobertaModel",
          "XLMRobertaPreTrainedModel",
      ]

try:
      if not is_tf_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase			:       List[str]									=				[
          "TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST",
          "TFXLMRobertaForCausalLM",
          "TFXLMRobertaForMaskedLM",
          "TFXLMRobertaForMultipleChoice",
          "TFXLMRobertaForQuestionAnswering",
          "TFXLMRobertaForSequenceClassification",
          "TFXLMRobertaForTokenClassification",
          "TFXLMRobertaModel",
          "TFXLMRobertaPreTrainedModel",
      ]

try:
      if not is_flax_available():
            raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
      pass
else:
      UpperCamelCase			:       List[str]									=				[
          "FLAX_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST",
          "FlaxXLMRobertaForMaskedLM",
          "FlaxXLMRobertaForCausalLM",
          "FlaxXLMRobertaForMultipleChoice",
          "FlaxXLMRobertaForQuestionAnswering",
          "FlaxXLMRobertaForSequenceClassification",
          "FlaxXLMRobertaForTokenClassification",
          "FlaxXLMRobertaModel",
          "FlaxXLMRobertaPreTrainedModel",
      ]

if TYPE_CHECKING:
      from .configuration_xlm_roberta import (
          XLM_ROBERTA_PRETRAINED_CONFIG_ARCHIVE_MAP,
          XLMRobertaConfig,
          XLMRobertaOnnxConfig,
      )

      try:
            if not is_sentencepiece_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .tokenization_xlm_roberta import XLMRobertaTokenizer

      try:
            if not is_tokenizers_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .tokenization_xlm_roberta_fast import XLMRobertaTokenizerFast

      try:
            if not is_torch_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_xlm_roberta import (
                XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
                XLMRobertaForCausalLM,
                XLMRobertaForMaskedLM,
                XLMRobertaForMultipleChoice,
                XLMRobertaForQuestionAnswering,
                XLMRobertaForSequenceClassification,
                XLMRobertaForTokenClassification,
                XLMRobertaModel,
                XLMRobertaPreTrainedModel,
            )

      try:
            if not is_tf_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_tf_xlm_roberta import (
                TF_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
                TFXLMRobertaForCausalLM,
                TFXLMRobertaForMaskedLM,
                TFXLMRobertaForMultipleChoice,
                TFXLMRobertaForQuestionAnswering,
                TFXLMRobertaForSequenceClassification,
                TFXLMRobertaForTokenClassification,
                TFXLMRobertaModel,
                TFXLMRobertaPreTrainedModel,
            )

      try:
            if not is_flax_available():
                  raise OptionalDependencyNotAvailable()
      except OptionalDependencyNotAvailable:
            pass
      else:
            from .modeling_flax_xlm_roberta import (
                FLAX_XLM_ROBERTA_PRETRAINED_MODEL_ARCHIVE_LIST,
                FlaxXLMRobertaForCausalLM,
                FlaxXLMRobertaForMaskedLM,
                FlaxXLMRobertaForMultipleChoice,
                FlaxXLMRobertaForQuestionAnswering,
                FlaxXLMRobertaForSequenceClassification,
                FlaxXLMRobertaForTokenClassification,
                FlaxXLMRobertaModel,
                FlaxXLMRobertaPreTrainedModel,
            )

else:
      import sys

      UpperCamelCase			:       Dict									=				_LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)



