






# Copyright 2023 The HuggingFace Inc. 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 ..models.auto import AutoModelForSeqaSeqLM, AutoTokenizer
from .base import PipelineTool
class 		SCREAMING_SNAKE_CASE__	(  UpperCAmelCase     ):

  '''simple docstring'''







  _lowerCamelCase    =  '''philschmid/bart-large-cnn-samsum'''
  _lowerCamelCase    =  (
      '''This is a tool that summarizes an English text. It takes an input `text` containing the text to summarize, '''
      '''and returns a summary of the text.'''
  )
  _lowerCamelCase    =  '''summarizer'''
  _lowerCamelCase    =  AutoTokenizer
  _lowerCamelCase    =  AutoModelForSeqaSeqLM

  _lowerCamelCase    =  ['''text''']
  _lowerCamelCase    =  ['''text''']


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


         """simple docstring"""
         return self.pre_processor(lowerCamelCase       ,				return_tensors="""pt"""       ,				truncation=lowerCamelCase							)


  def  lowerCamelCase       (      self		:			Any       ,				lowerCamelCase		:			Optional[Any]							)						->	int:


         """simple docstring"""
         return self.model.generate(**lowerCamelCase							)[0]





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


         """simple docstring"""
         return self.pre_processor.decode(lowerCamelCase       ,				skip_special_tokens=lowerCamelCase       ,				clean_up_tokenization_spaces=lowerCamelCase							)







import os
import sys
import unittest


__a:     Optional[int]   					=					os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
sys.path.append(os.path.join(git_repo_path, '''utils'''))

import get_test_info  # noqa: E402
from get_test_info import (  # noqa: E402
    get_model_to_test_mapping,
    get_model_to_tester_mapping,
    get_test_to_tester_mapping,
)


__a:     Dict   					=					os.path.join('''tests''', '''models''', '''bert''', '''test_modeling_bert.py''')
__a:     Union[str, Any]   					=					os.path.join('''tests''', '''models''', '''blip''', '''test_modeling_blip.py''')
class 		SCREAMING_SNAKE_CASE__	(  unittest.TestCase     ):

  '''simple docstring'''







  def  lowerCamelCase       (      self		:			str							)						->	Dict:


         """simple docstring"""
         _UpperCAmelCase							=    get_test_to_tester_mapping(lowerCamelCase							)
         _UpperCAmelCase							=    get_test_to_tester_mapping(lowerCamelCase							)

         _UpperCAmelCase							=    {"""BertModelTest""": """BertModelTester"""}

         _UpperCAmelCase							=    {
             """BlipModelTest""": """BlipModelTester""",
             """BlipTextImageModelTest""": """BlipTextImageModelsModelTester""",
             """BlipTextModelTest""": """BlipTextModelTester""",
             """BlipTextRetrievalModelTest""": """BlipTextRetrievalModelTester""",
             """BlipVQAModelTest""": """BlipVQAModelTester""",
             """BlipVisionModelTest""": """BlipVisionModelTester""",
         }

         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)
         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)


  def  lowerCamelCase       (      self		:			Optional[int]							)						->	str:


         """simple docstring"""
         _UpperCAmelCase							=    get_model_to_test_mapping(lowerCamelCase							)
         _UpperCAmelCase							=    get_model_to_test_mapping(lowerCamelCase							)

         _UpperCAmelCase							=    {
             """BertForMaskedLM""": ["""BertModelTest"""],
             """BertForMultipleChoice""": ["""BertModelTest"""],
             """BertForNextSentencePrediction""": ["""BertModelTest"""],
             """BertForPreTraining""": ["""BertModelTest"""],
             """BertForQuestionAnswering""": ["""BertModelTest"""],
             """BertForSequenceClassification""": ["""BertModelTest"""],
             """BertForTokenClassification""": ["""BertModelTest"""],
             """BertLMHeadModel""": ["""BertModelTest"""],
             """BertModel""": ["""BertModelTest"""],
         }

         _UpperCAmelCase							=    {
             """BlipForConditionalGeneration""": ["""BlipTextImageModelTest"""],
             """BlipForImageTextRetrieval""": ["""BlipTextRetrievalModelTest"""],
             """BlipForQuestionAnswering""": ["""BlipVQAModelTest"""],
             """BlipModel""": ["""BlipModelTest"""],
             """BlipTextModel""": ["""BlipTextModelTest"""],
             """BlipVisionModel""": ["""BlipVisionModelTest"""],
         }

         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)
         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)





  def  lowerCamelCase       (      self		:			int							)						->	Any:


         """simple docstring"""
         _UpperCAmelCase							=    get_model_to_tester_mapping(lowerCamelCase							)
         _UpperCAmelCase							=    get_model_to_tester_mapping(lowerCamelCase							)

         _UpperCAmelCase							=    {
             """BertForMaskedLM""": ["""BertModelTester"""],
             """BertForMultipleChoice""": ["""BertModelTester"""],
             """BertForNextSentencePrediction""": ["""BertModelTester"""],
             """BertForPreTraining""": ["""BertModelTester"""],
             """BertForQuestionAnswering""": ["""BertModelTester"""],
             """BertForSequenceClassification""": ["""BertModelTester"""],
             """BertForTokenClassification""": ["""BertModelTester"""],
             """BertLMHeadModel""": ["""BertModelTester"""],
             """BertModel""": ["""BertModelTester"""],
         }

         _UpperCAmelCase							=    {
             """BlipForConditionalGeneration""": ["""BlipTextImageModelsModelTester"""],
             """BlipForImageTextRetrieval""": ["""BlipTextRetrievalModelTester"""],
             """BlipForQuestionAnswering""": ["""BlipVQAModelTester"""],
             """BlipModel""": ["""BlipModelTester"""],
             """BlipTextModel""": ["""BlipTextModelTester"""],
             """BlipVisionModel""": ["""BlipVisionModelTester"""],
         }

         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)
         self.assertEqual(get_test_info.to_json(lowerCamelCase							)       ,				lowerCamelCase							)