




'''simple docstring'''





import enum
import inspect
import warnings
from functools import wraps
from typing import Callable, Optional

from .logging import get_logger


lowercase__								=						set()
lowercase__								=						get_logger(__name__)


def 					UpperCamelCase						(	snake_case = None ):




   '''simple docstring'''

   def decorator(snake_case ):
      global _emitted_deprecation_warnings

      if inspect.isclass(snake_case ):
         SCREAMING_SNAKE_CASE__      :					Optional[int]							    =			deprecated_class_or_function.__init__
         SCREAMING_SNAKE_CASE__      :					Tuple							    =			deprecated_class_or_function.__name__
      else:
         SCREAMING_SNAKE_CASE__      :					Any							    =			deprecated_class_or_function
         SCREAMING_SNAKE_CASE__      :					Union[str, Any]							    =			deprecated_function.__name__
         # Support deprecating __init__ class method: class name instead
         SCREAMING_SNAKE_CASE__      :					Optional[Any]							    =			name if name != '''__init__''' else deprecated_function.__qualname__.split('''.''' )[-2]

      SCREAMING_SNAKE_CASE__      :					Optional[Any]							    =			(
          f"""{name} is deprecated and will be removed in the next major version of datasets.""" + f""" {help_message}"""
          if help_message
          else ''''''
      )

      @wraps(snake_case )
      def wrapper(*snake_case						,  **snake_case ):
         SCREAMING_SNAKE_CASE__      :					Optional[Any]							    =			hash(snake_case )
         if func_hash not in _emitted_deprecation_warnings:
            warnings.warn(snake_case						,  category=snake_case						,  stacklevel=2 )
            _emitted_deprecation_warnings.add(snake_case )
         return deprecated_function(*snake_case						,  **snake_case )

      SCREAMING_SNAKE_CASE__      :					Any							    =			'''deprecated'''

      if inspect.isclass(snake_case ):
         SCREAMING_SNAKE_CASE__      :					Optional[int]							    =			wrapper
         return deprecated_class_or_function
      else:
         return wrapper

   return decorator






class     lowercase  ( enum.EnumMeta):






      def __getattribute__(cls			,	a__       )							-> str:
         SCREAMING_SNAKE_CASE__      :					int							    =			super().__getattribute__(a__       )
         if isinstance(a__			,	enum.Enum       ) and obj._on_access:
            obj._on_access()
         return obj






      def __getitem__(cls			,	a__       )							-> Tuple:
         SCREAMING_SNAKE_CASE__      :					Dict							    =			super().__getitem__(a__       )
         if member._on_access:
            member._on_access()
         return member






      def __call__(cls			,	a__			,	a__=None			,	*, a__=None			,	a__=None			,	a__=None			,	a__=1       )							-> Optional[int]:
         SCREAMING_SNAKE_CASE__      :					Optional[Any]							    =			super().__call__(a__			,	a__			,	module=a__			,	qualname=a__			,	type=a__			,	start=a__       )
         if isinstance(a__			,	enum.Enum       ) and obj._on_access:
            obj._on_access()
         return obj






class     lowercase  ( enum.Enum					,      metaclass=UpperCAmelCase_):






      def __new__(cls			,	a__       )							-> Any:
         SCREAMING_SNAKE_CASE__      :					Dict							    =			object.__new__(cls       )
         SCREAMING_SNAKE_CASE__      :					Any							    =			value
         SCREAMING_SNAKE_CASE__      :					int							    =			member.deprecate
         return member






      @property
      def  A__(self       )							-> Dict:
         return ""






      def  A__(self       )							-> List[Any]:
         SCREAMING_SNAKE_CASE__      :					str							    =			f""" {self.help_message}""" if self.help_message else ''''''
         warnings.warn(
             f"""'{self.__objclass__.__name__}' is deprecated and will be removed in the next major version of datasets."""
             + help_message			,	a__			,	stacklevel=3			,	)







'''simple docstring'''





import unittest

from transformers import (
    MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING,
    TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING,
    PreTrainedTokenizer,
    is_vision_available,
)
from transformers.pipelines import ImageClassificationPipeline, pipeline
from transformers.testing_utils import (
    is_pipeline_test,
    nested_simplify,
    require_tf,
    require_torch,
    require_torch_or_tf,
    require_vision,
    slow,
)

from .test_pipelines_common import ANY


if is_vision_available():
       from PIL import Image
else:






       class     lowercase  :






             @staticmethod
             def  A__(*a__			,	**a__       )							-> Union[str, Any]:
                pass






@is_pipeline_test
@require_torch_or_tf
@require_vision
class     lowercase  ( unittest.TestCase):
      __lowercase					=	MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING
      __lowercase					=	TF_MODEL_FOR_IMAGE_CLASSIFICATION_MAPPING






      def  A__(self			,	a__			,	a__			,	a__       )							-> Dict:
         SCREAMING_SNAKE_CASE__      :					Optional[int]							    =			ImageClassificationPipeline(model=a__			,	image_processor=a__			,	top_k=2       )
         SCREAMING_SNAKE_CASE__      :					List[Any]							    =			[
             Image.open('''./tests/fixtures/tests_samples/COCO/000000039769.png'''       ),
             '''http://images.cocodataset.org/val2017/000000039769.jpg''',
         ]
         return image_classifier, examples






      def  A__(self			,	a__			,	a__       )							-> List[Any]:
         SCREAMING_SNAKE_CASE__      :					int							    =			image_classifier('''./tests/fixtures/tests_samples/COCO/000000039769.png'''       )

         self.assertEqual(
             a__			,	[
                 {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                 {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
             ]			,	)

         import datasets

         SCREAMING_SNAKE_CASE__      :					Optional[Any]							    =			datasets.load_dataset('''hf-internal-testing/fixtures_image_utils'''			,	'''image'''			,	split='''test'''       )

         # Accepts URL + PIL.Image + lists
         SCREAMING_SNAKE_CASE__      :					Tuple							    =			image_classifier(
             [
                 Image.open('''./tests/fixtures/tests_samples/COCO/000000039769.png'''       ),
                 '''http://images.cocodataset.org/val2017/000000039769.jpg''',
                 # RGBA
                 dataset[0]['''file'''],
                 # LA
                 dataset[1]['''file'''],
                 # L
                 dataset[2]['''file'''],
             ]       )
         self.assertEqual(
             a__			,	[
                 [
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                 ],
                 [
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                 ],
                 [
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                 ],
                 [
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                 ],
                 [
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                     {'''score''': ANY(a__       ), '''label''': ANY(a__       )},
                 ],
             ]			,	)






      @require_torch
      def  A__(self       )							-> List[str]:
         SCREAMING_SNAKE_CASE__      :					List[str]							    =			'''hf-internal-testing/tiny-random-vit'''
         SCREAMING_SNAKE_CASE__      :					Any							    =			pipeline('''image-classification'''			,	model=a__       )

         SCREAMING_SNAKE_CASE__      :					Any							    =			image_classifier('''http://images.cocodataset.org/val2017/000000039769.jpg'''       )
         self.assertEqual(
             nested_simplify(a__			,	decimals=4       )			,	[{'''label''': '''LABEL_1''', '''score''': 0.5_74}, {'''label''': '''LABEL_0''', '''score''': 0.4_26}]			,	)

         SCREAMING_SNAKE_CASE__      :					Any							    =			image_classifier(
             [
                 '''http://images.cocodataset.org/val2017/000000039769.jpg''',
                 '''http://images.cocodataset.org/val2017/000000039769.jpg''',
             ]			,	top_k=2			,	)
         self.assertEqual(
             nested_simplify(a__			,	decimals=4       )			,	[
                 [{'''label''': '''LABEL_1''', '''score''': 0.5_74}, {'''label''': '''LABEL_0''', '''score''': 0.4_26}],
                 [{'''label''': '''LABEL_1''', '''score''': 0.5_74}, {'''label''': '''LABEL_0''', '''score''': 0.4_26}],
             ]			,	)






      @require_tf
      def  A__(self       )							-> Optional[int]:
         SCREAMING_SNAKE_CASE__      :					Optional[Any]							    =			'''hf-internal-testing/tiny-random-vit'''
         SCREAMING_SNAKE_CASE__      :					Optional[Any]							    =			pipeline('''image-classification'''			,	model=a__			,	framework='''tf'''       )

         SCREAMING_SNAKE_CASE__      :					List[str]							    =			image_classifier('''http://images.cocodataset.org/val2017/000000039769.jpg'''       )
         self.assertEqual(
             nested_simplify(a__			,	decimals=4       )			,	[{'''label''': '''LABEL_1''', '''score''': 0.5_74}, {'''label''': '''LABEL_0''', '''score''': 0.4_26}]			,	)

         SCREAMING_SNAKE_CASE__      :					Dict							    =			image_classifier(
             [
                 '''http://images.cocodataset.org/val2017/000000039769.jpg''',
                 '''http://images.cocodataset.org/val2017/000000039769.jpg''',
             ]			,	top_k=2			,	)
         self.assertEqual(
             nested_simplify(a__			,	decimals=4       )			,	[
                 [{'''label''': '''LABEL_1''', '''score''': 0.5_74}, {'''label''': '''LABEL_0''', '''score''': 0.4_26}],
                 [{'''label''': '''LABEL_1''', '''score''': 0.5_74}, {'''label''': '''LABEL_0''', '''score''': 0.4_26}],
             ]			,	)






      def  A__(self       )							-> Union[str, Any]:
         SCREAMING_SNAKE_CASE__      :					Optional[Any]							    =			PreTrainedTokenizer()

         # Assert that the pipeline can be initialized with a feature extractor that is not in any mapping
         SCREAMING_SNAKE_CASE__      :					Optional[int]							    =			pipeline(
             '''image-classification'''			,	model='''hf-internal-testing/tiny-random-vit'''			,	tokenizer=a__       )

         self.assertIs(image_classifier.tokenizer			,	a__       )






      @slow
      @require_torch
      def  A__(self       )							-> int:
         # Perceiver is not tested by `run_pipeline_test` properly.
         # That is because the type of feature_extractor and model preprocessor need to be kept
         # in sync, which is not the case in the current design
         SCREAMING_SNAKE_CASE__      :					Union[str, Any]							    =			pipeline('''image-classification'''			,	model='''deepmind/vision-perceiver-conv'''       )
         SCREAMING_SNAKE_CASE__      :					str							    =			image_classifier('''http://images.cocodataset.org/val2017/000000039769.jpg'''       )
         self.assertEqual(
             nested_simplify(a__			,	decimals=4       )			,	[
                 {'''score''': 0.43_85, '''label''': '''tabby, tabby cat'''},
                 {'''score''': 0.3_21, '''label''': '''tiger cat'''},
                 {'''score''': 0.05_02, '''label''': '''Egyptian cat'''},
                 {'''score''': 0.01_37, '''label''': '''crib, cot'''},
                 {'''score''': 0.0_07, '''label''': '''radiator'''},
             ]			,	)

         SCREAMING_SNAKE_CASE__      :					str							    =			pipeline('''image-classification'''			,	model='''deepmind/vision-perceiver-fourier'''       )
         SCREAMING_SNAKE_CASE__      :					Optional[Any]							    =			image_classifier('''http://images.cocodataset.org/val2017/000000039769.jpg'''       )
         self.assertEqual(
             nested_simplify(a__			,	decimals=4       )			,	[
                 {'''score''': 0.56_58, '''label''': '''tabby, tabby cat'''},
                 {'''score''': 0.13_09, '''label''': '''tiger cat'''},
                 {'''score''': 0.07_22, '''label''': '''Egyptian cat'''},
                 {'''score''': 0.07_07, '''label''': '''remote control, remote'''},
                 {'''score''': 0.00_82, '''label''': '''computer keyboard, keypad'''},
             ]			,	)

         SCREAMING_SNAKE_CASE__      :					Dict							    =			pipeline('''image-classification'''			,	model='''deepmind/vision-perceiver-learned'''       )
         SCREAMING_SNAKE_CASE__      :					Optional[int]							    =			image_classifier('''http://images.cocodataset.org/val2017/000000039769.jpg'''       )
         self.assertEqual(
             nested_simplify(a__			,	decimals=4       )			,	[
                 {'''score''': 0.30_22, '''label''': '''tabby, tabby cat'''},
                 {'''score''': 0.23_62, '''label''': '''Egyptian cat'''},
                 {'''score''': 0.18_56, '''label''': '''tiger cat'''},
                 {'''score''': 0.03_24, '''label''': '''remote control, remote'''},
                 {'''score''': 0.00_96, '''label''': '''quilt, comforter, comfort, puff'''},
             ]			,	)

