






# 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.
import torch

from ..models.auto import AutoModelForSequenceClassification, AutoTokenizer
from .base import PipelineTool






class 			_UpperCamelCase(       __lowerCamelCase		):
		__SCREAMING_SNAKE_CASE  :       List[str]							=							'''facebook/bart-large-mnli'''
		__SCREAMING_SNAKE_CASE  :       int							=							(
		    '''This is a tool that classifies an English text using provided labels. It takes two inputs: `text`, which '''
		    '''should be the text to classify, and `labels`, which should be the list of labels to use for classification. '''
		    '''It returns the most likely label in the list of provided `labels` for the input text.'''
		)
		__SCREAMING_SNAKE_CASE  :       Optional[Any]							=							'''text_classifier'''
		__SCREAMING_SNAKE_CASE  :       Any							=							AutoTokenizer
		__SCREAMING_SNAKE_CASE  :       Optional[int]							=							AutoModelForSequenceClassification

		__SCREAMING_SNAKE_CASE  :       List[str]							=							['''text''', ['''text''']]
		__SCREAMING_SNAKE_CASE  :       Optional[int]							=							['''text''']

		def    __lowerCAmelCase  (					self		:    Optional[int]    ):



					'''simple docstring'''



					super().setup()
					__a						:  Tuple									=     self.model.config
					__a						:  str									=     -1
					for idx, label in config.idalabel.items():
								if label.lower().startswith('entail'    ):
											__a						:  Any									=     int(SCREAMING_SNAKE_CASE__    )
					if self.entailment_id == -1:
								raise ValueError('Could not determine the entailment ID from the model config, please pass it at init.'    )

		def    __lowerCAmelCase  (					self		:    int    ,       SCREAMING_SNAKE_CASE__		:    str    ,       SCREAMING_SNAKE_CASE__		:    Tuple    ):



					'''simple docstring'''



					__a						:  Any									=     labels
					return self.pre_processor(
					    [text] * len(SCREAMING_SNAKE_CASE__    )    ,       [f'''This example is {label}''' for label in labels]    ,       return_tensors='pt'    ,       padding='max_length'    ,       )



		def    __lowerCAmelCase  (					self		:    Tuple    ,       SCREAMING_SNAKE_CASE__		:    Tuple    ):



					'''simple docstring'''



					__a						:  Dict									=     outputs.logits
					__a						:  List[str]									=     torch.argmax(logits[:, 2]    ).item()
					return self._labels[label_id]











from typing import List, Optional, TypeVar

from .arrow_dataset import Dataset, _concatenate_map_style_datasets, _interleave_map_style_datasets
from .dataset_dict import DatasetDict, IterableDatasetDict
from .info import DatasetInfo
from .iterable_dataset import IterableDataset, _concatenate_iterable_datasets, _interleave_iterable_datasets
from .splits import NamedSplit
from .utils import logging
from .utils.py_utils import Literal


UpperCamelCase__ :   Optional[Any]						=							logging.get_logger(__name__)


UpperCamelCase__ :   int						=							TypeVar('''DatasetType''', Dataset, IterableDataset)





def        __UpperCAmelCase						(							lowerCamelCase_ :     List[DatasetType]  ,							lowerCamelCase_ :     Optional[List[float]] = None  ,							lowerCamelCase_ :     Optional[int] = None  ,							lowerCamelCase_ :     Optional[DatasetInfo] = None  ,							lowerCamelCase_ :     Optional[NamedSplit] = None  ,							lowerCamelCase_ :     Literal["first_exhausted", "all_exhausted"] = "first_exhausted"  ,							)       ->  DatasetType:







			"""simple docstring"""





			from .arrow_dataset import Dataset
			from .iterable_dataset import IterableDataset

			if not datasets:
						raise ValueError('Unable to interleave an empty list of datasets.'  )
			for i, dataset in enumerate(lowerCamelCase_  ):
						if not isinstance(lowerCamelCase_  ,							(Dataset, IterableDataset)  ):
									if isinstance(lowerCamelCase_  ,							(DatasetDict, IterableDatasetDict)  ):
												if not dataset:
															raise ValueError(
															    F'Expected a list of Dataset objects or a list of IterableDataset objects, but element at position {i} '
															    'is an empty dataset dictionary.'  )
												raise ValueError(
												    F'Dataset at position {i} has at least one split: {list(lowerCamelCase_  )}\n'
												    F'Please pick one to interleave with the other datasets, for example: dataset[\'{next(iter(lowerCamelCase_  )  )}\']'  )
									raise ValueError(
									    F'Expected a list of Dataset objects or a list of IterableDataset objects, but element at position {i} is a {type(lowerCamelCase_  ).__name__}.'  )
						if i == 0:
									SCREAMING_SNAKE_CASE_  ,    SCREAMING_SNAKE_CASE_   : Dict					  =      (
									    (Dataset, IterableDataset) if isinstance(lowerCamelCase_  ,							lowerCamelCase_  ) else (IterableDataset, Dataset)
									)
						elif not isinstance(lowerCamelCase_  ,							lowerCamelCase_  ):
									raise ValueError(
									    F'Unable to interleave a {dataset_type.__name__} (at position 0) with a {other_type.__name__} (at position {i}). Expected a list of Dataset objects or a list of IterableDataset objects.'  )
			if stopping_strategy not in ["first_exhausted", "all_exhausted"]:
						raise ValueError(F'{stopping_strategy} is not supported. Please enter a valid stopping_strategy.'  )
			if dataset_type is Dataset:
						return _interleave_map_style_datasets(
						    lowerCamelCase_  ,							lowerCamelCase_  ,							lowerCamelCase_  ,							info=lowerCamelCase_  ,							split=lowerCamelCase_  ,							stopping_strategy=lowerCamelCase_  )
			else:
						return _interleave_iterable_datasets(
						    lowerCamelCase_  ,							lowerCamelCase_  ,							lowerCamelCase_  ,							info=lowerCamelCase_  ,							split=lowerCamelCase_  ,							stopping_strategy=lowerCamelCase_  )





def        __UpperCAmelCase						(							lowerCamelCase_ :     List[DatasetType]  ,							lowerCamelCase_ :     Optional[DatasetInfo] = None  ,							lowerCamelCase_ :     Optional[NamedSplit] = None  ,							lowerCamelCase_ :     int = 0  ,							)       ->  DatasetType:







			"""simple docstring"""





			if not dsets:
						raise ValueError('Unable to concatenate an empty list of datasets.'  )
			for i, dataset in enumerate(lowerCamelCase_  ):
						if not isinstance(lowerCamelCase_  ,							(Dataset, IterableDataset)  ):
									if isinstance(lowerCamelCase_  ,							(DatasetDict, IterableDatasetDict)  ):
												if not dataset:
															raise ValueError(
															    F'Expected a list of Dataset objects or a list of IterableDataset objects, but element at position {i} '
															    'is an empty dataset dictionary.'  )
												raise ValueError(
												    F'Dataset at position {i} has at least one split: {list(lowerCamelCase_  )}\n'
												    F'Please pick one to interleave with the other datasets, for example: dataset[\'{next(iter(lowerCamelCase_  )  )}\']'  )
									raise ValueError(
									    F'Expected a list of Dataset objects or a list of IterableDataset objects, but element at position {i} is a {type(lowerCamelCase_  ).__name__}.'  )
						if i == 0:
									SCREAMING_SNAKE_CASE_  ,    SCREAMING_SNAKE_CASE_   : Any					  =      (
									    (Dataset, IterableDataset) if isinstance(lowerCamelCase_  ,							lowerCamelCase_  ) else (IterableDataset, Dataset)
									)
						elif not isinstance(lowerCamelCase_  ,							lowerCamelCase_  ):
									raise ValueError(
									    F'Unable to interleave a {dataset_type.__name__} (at position 0) with a {other_type.__name__} (at position {i}). Expected a list of Dataset objects or a list of IterableDataset objects.'  )
			if dataset_type is Dataset:
						return _concatenate_map_style_datasets(lowerCamelCase_  ,							info=lowerCamelCase_  ,							split=lowerCamelCase_  ,							axis=lowerCamelCase_  )
			else:
						return _concatenate_iterable_datasets(lowerCamelCase_  ,							info=lowerCamelCase_  ,							split=lowerCamelCase_  ,							axis=lowerCamelCase_  )



