





# Copyright 2023 The HuggingFace Team. All rights reserved.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
#     http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_torch_available


UpperCAmelCase	:	Tuple		     =       {
    """configuration_xmod""": [
        """XMOD_PRETRAINED_CONFIG_ARCHIVE_MAP""",
        """XmodConfig""",
        """XmodOnnxConfig""",
    ],
}

try:
     if not is_torch_available():
          raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
     pass
else:
     UpperCAmelCase	:	Dict		     =       [
         """XMOD_PRETRAINED_MODEL_ARCHIVE_LIST""",
         """XmodForCausalLM""",
         """XmodForMaskedLM""",
         """XmodForMultipleChoice""",
         """XmodForQuestionAnswering""",
         """XmodForSequenceClassification""",
         """XmodForTokenClassification""",
         """XmodModel""",
         """XmodPreTrainedModel""",
     ]

if TYPE_CHECKING:
     from .configuration_xmod import XMOD_PRETRAINED_CONFIG_ARCHIVE_MAP, XmodConfig, XmodOnnxConfig

     try:
          if not is_torch_available():
               raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
          pass
     else:
          from .modeling_xmod import (
              XMOD_PRETRAINED_MODEL_ARCHIVE_LIST,
              XmodForCausalLM,
              XmodForMaskedLM,
              XmodForMultipleChoice,
              XmodForQuestionAnswering,
              XmodForSequenceClassification,
              XmodForTokenClassification,
              XmodModel,
              XmodPreTrainedModel,
          )

else:
     import sys

     UpperCAmelCase	:	int		     =       _LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)









"""simple docstring"""





import unittest

from huggingface_hub import hf_hub_download

from transformers import MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING, VideoMAEFeatureExtractor
from transformers.pipelines import VideoClassificationPipeline, pipeline
from transformers.testing_utils import (
    is_pipeline_test,
    nested_simplify,
    require_decord,
    require_tf,
    require_torch,
    require_torch_or_tf,
    require_vision,
)

from .test_pipelines_common import ANY







@is_pipeline_test
@require_torch_or_tf
@require_vision
@require_decord
class 				SCREAMING_SNAKE_CASE					(		unittest.TestCase       ):






					"""simple docstring"""



					lowercase__				 =       MODEL_FOR_VIDEO_CLASSIFICATION_MAPPING







					def        __lowerCAmelCase (       self			:				Optional[Any]   ,lowercase_			:				Optional[Any]   ,lowercase_			:				int   ,lowercase_			:				str		):
						lowerCAmelCase__	:						Optional[int]  				=	hf_hub_download(
						    repo_id='''nateraw/video-demo'''   ,filename='''archery.mp4'''   ,repo_type='''dataset'''		)
						lowerCAmelCase__	:						Tuple  				=	VideoClassificationPipeline(model=lowercase_   ,image_processor=lowercase_   ,top_k=2		)
						lowerCAmelCase__	:						Optional[int]  				=	[
						    example_video_filepath,
						    '''https://huggingface.co/datasets/nateraw/video-demo/resolve/main/archery.mp4''',
						]
						return video_classifier, examples







					def        __lowerCAmelCase (       self			:				str   ,lowercase_			:				int   ,lowercase_			:				Dict		):
						for example in examples:
							lowerCAmelCase__	:						Dict  				=	video_classifier(lowercase_		)

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







					@require_torch
					def        __lowerCAmelCase (       self			:				Optional[int]		):
						lowerCAmelCase__	:						Optional[int]  				=	'''hf-internal-testing/tiny-random-VideoMAEForVideoClassification'''
						lowerCAmelCase__	:						List[str]  				=	VideoMAEFeatureExtractor(
						    size={'''shortest_edge''': 1_0}   ,crop_size={'''height''': 1_0, '''width''': 1_0}		)
						lowerCAmelCase__	:						Optional[Any]  				=	pipeline(
						    '''video-classification'''   ,model=lowercase_   ,feature_extractor=lowercase_   ,frame_sampling_rate=4		)

						lowerCAmelCase__	:						Optional[int]  				=	hf_hub_download(repo_id='''nateraw/video-demo'''   ,filename='''archery.mp4'''   ,repo_type='''dataset'''		)
						lowerCAmelCase__	:						Optional[int]  				=	video_classifier(lowercase_   ,top_k=2		)
						self.assertEqual(
						    nested_simplify(lowercase_   ,decimals=4		)   ,[{'''score''': 0.5199, '''label''': '''LABEL_0'''}, {'''score''': 0.4801, '''label''': '''LABEL_1'''}]   ,)

						lowerCAmelCase__	:						Dict  				=	video_classifier(
						    [
						        video_file_path,
						        video_file_path,
						    ]   ,top_k=2   ,)
						self.assertEqual(
						    nested_simplify(lowercase_   ,decimals=4		)   ,[
						        [{'''score''': 0.5199, '''label''': '''LABEL_0'''}, {'''score''': 0.4801, '''label''': '''LABEL_1'''}],
						        [{'''score''': 0.5199, '''label''': '''LABEL_0'''}, {'''score''': 0.4801, '''label''': '''LABEL_1'''}],
						    ]   ,)







					@require_tf
					def        __lowerCAmelCase (       self			:				int		):
						pass




