"""simple docstring"""





import json
from typing import List, Optional, Tuple

from tokenizers import normalizers

from ...tokenization_utils_fast import PreTrainedTokenizerFast
from ...utils import logging
from .tokenization_funnel import FunnelTokenizer


lowerCAmelCase_     =	logging.get_logger(__name__)

lowerCAmelCase_     =	{'''vocab_file''': '''vocab.txt''', '''tokenizer_file''': '''tokenizer.json'''}

lowerCAmelCase_     =	[
    '''small''',
    '''small-base''',
    '''medium''',
    '''medium-base''',
    '''intermediate''',
    '''intermediate-base''',
    '''large''',
    '''large-base''',
    '''xlarge''',
    '''xlarge-base''',
]

lowerCAmelCase_     =	{
    '''vocab_file''': {
        '''funnel-transformer/small''': '''https://huggingface.co/funnel-transformer/small/resolve/main/vocab.txt''',
        '''funnel-transformer/small-base''': '''https://huggingface.co/funnel-transformer/small-base/resolve/main/vocab.txt''',
        '''funnel-transformer/medium''': '''https://huggingface.co/funnel-transformer/medium/resolve/main/vocab.txt''',
        '''funnel-transformer/medium-base''': (
            '''https://huggingface.co/funnel-transformer/medium-base/resolve/main/vocab.txt'''
        ),
        '''funnel-transformer/intermediate''': (
            '''https://huggingface.co/funnel-transformer/intermediate/resolve/main/vocab.txt'''
        ),
        '''funnel-transformer/intermediate-base''': (
            '''https://huggingface.co/funnel-transformer/intermediate-base/resolve/main/vocab.txt'''
        ),
        '''funnel-transformer/large''': '''https://huggingface.co/funnel-transformer/large/resolve/main/vocab.txt''',
        '''funnel-transformer/large-base''': '''https://huggingface.co/funnel-transformer/large-base/resolve/main/vocab.txt''',
        '''funnel-transformer/xlarge''': '''https://huggingface.co/funnel-transformer/xlarge/resolve/main/vocab.txt''',
        '''funnel-transformer/xlarge-base''': (
            '''https://huggingface.co/funnel-transformer/xlarge-base/resolve/main/vocab.txt'''
        ),
    },
    '''tokenizer_file''': {
        '''funnel-transformer/small''': '''https://huggingface.co/funnel-transformer/small/resolve/main/tokenizer.json''',
        '''funnel-transformer/small-base''': (
            '''https://huggingface.co/funnel-transformer/small-base/resolve/main/tokenizer.json'''
        ),
        '''funnel-transformer/medium''': '''https://huggingface.co/funnel-transformer/medium/resolve/main/tokenizer.json''',
        '''funnel-transformer/medium-base''': (
            '''https://huggingface.co/funnel-transformer/medium-base/resolve/main/tokenizer.json'''
        ),
        '''funnel-transformer/intermediate''': (
            '''https://huggingface.co/funnel-transformer/intermediate/resolve/main/tokenizer.json'''
        ),
        '''funnel-transformer/intermediate-base''': (
            '''https://huggingface.co/funnel-transformer/intermediate-base/resolve/main/tokenizer.json'''
        ),
        '''funnel-transformer/large''': '''https://huggingface.co/funnel-transformer/large/resolve/main/tokenizer.json''',
        '''funnel-transformer/large-base''': (
            '''https://huggingface.co/funnel-transformer/large-base/resolve/main/tokenizer.json'''
        ),
        '''funnel-transformer/xlarge''': '''https://huggingface.co/funnel-transformer/xlarge/resolve/main/tokenizer.json''',
        '''funnel-transformer/xlarge-base''': (
            '''https://huggingface.co/funnel-transformer/xlarge-base/resolve/main/tokenizer.json'''
        ),
    },
}
lowerCAmelCase_     =	{F"funnel-transformer/{name}": 512 for name in _model_names}
lowerCAmelCase_     =	{F"funnel-transformer/{name}": {'''do_lower_case''': True} for name in _model_names}




class  _snake_case		(    __snake_case ):







				"""simple docstring"""






				a			=						VOCAB_FILES_NAMES
				a			=						PRETRAINED_VOCAB_FILES_MAP
				a			=						PRETRAINED_INIT_CONFIGURATION
				a			=						FunnelTokenizer
				a			=						PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
				a			=						2





				def __init__(      self							:							int      ,    _A							:							Union[str, Any]=None      ,    _A							:							Any=None      ,    _A							:							List[str]=True      ,    _A							:							Optional[Any]="<unk>"      ,    _A							:							Optional[int]="<sep>"      ,    _A							:							int="<pad>"      ,    _A							:							int="<cls>"      ,    _A							:							Optional[Any]="<mask>"      ,    _A							:							Tuple="<s>"      ,    _A							:							Optional[int]="</s>"      ,    _A							:							List[Any]=True      ,    _A							:							Optional[Any]=True      ,    _A							:							List[Any]=None      ,    _A							:							List[Any]="##"      ,    **_A							:							Optional[Any]      ,    ):



											"""simple docstring"""







											super().__init__(
											    _A      ,    tokenizer_file=_A      ,    do_lower_case=_A      ,    unk_token=_A      ,    sep_token=_A      ,    pad_token=_A      ,    cls_token=_A      ,    mask_token=_A      ,    bos_token=_A      ,    eos_token=_A      ,    clean_text=_A      ,    tokenize_chinese_chars=_A      ,    strip_accents=_A      ,    wordpieces_prefix=_A      ,    **_A      ,    )

											_SCREAMING_SNAKE_CASE		:						Dict			  =    json.loads(self.backend_tokenizer.normalizer.__getstate__())
											if (
											    normalizer_state.get("""lowercase"""      ,    _A) != do_lower_case
											    or normalizer_state.get("""strip_accents"""      ,    _A) != strip_accents
											    or normalizer_state.get("""handle_chinese_chars"""      ,    _A) != tokenize_chinese_chars
											):
																		_SCREAMING_SNAKE_CASE		:						List[str]			  =    getattr(_A      ,    normalizer_state.pop("""type"""))
																		_SCREAMING_SNAKE_CASE		:						Tuple			  =    do_lower_case
																		_SCREAMING_SNAKE_CASE		:						Optional[Any]			  =    strip_accents
																		_SCREAMING_SNAKE_CASE		:						Any			  =    tokenize_chinese_chars
																		_SCREAMING_SNAKE_CASE		:						Tuple			  =    normalizer_class(**_A)

											_SCREAMING_SNAKE_CASE		:						Dict			  =    do_lower_case





				def 		_lowerCAmelCase		(      self							:							int      ,    _A							:							Optional[int]      ,    _A							:							List[str]=None):



											"""simple docstring"""







											_SCREAMING_SNAKE_CASE		:						List[Any]			  =    [self.cls_token_id] + token_ids_a + [self.sep_token_id]

											if token_ids_a:
																		output += token_ids_a + [self.sep_token_id]

											return output





				def 		_lowerCAmelCase		(      self							:							Optional[int]      ,    _A							:							List[int]      ,    _A							:							Optional[List[int]] = None):



											"""simple docstring"""







											_SCREAMING_SNAKE_CASE		:						Tuple			  =    [self.sep_token_id]
											_SCREAMING_SNAKE_CASE		:						Optional[int]			  =    [self.cls_token_id]
											if token_ids_a is None:
																		return len(cls) * [self.cls_token_type_id] + len(token_ids_a + sep) * [0]
											return len(cls) * [self.cls_token_type_id] + len(token_ids_a + sep) * [0] + len(token_ids_a + sep) * [1]





				def 		_lowerCAmelCase		(      self							:							Tuple      ,    _A							:							str      ,    _A							:							Optional[str] = None):



											"""simple docstring"""







											_SCREAMING_SNAKE_CASE		:						Tuple			  =    self._tokenizer.model.save(_A      ,    name=_A)
											return tuple(_A)



"""simple docstring"""





import pyarrow.parquet as pq
import pytest

from datasets import Audio, Dataset, DatasetDict, Features, NamedSplit, Sequence, Value, config
from datasets.features.image import Image
from datasets.io.parquet import ParquetDatasetReader, ParquetDatasetWriter, get_writer_batch_size

from ..utils import assert_arrow_memory_doesnt_increase, assert_arrow_memory_increases



def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    Optional[int]:
							assert isinstance(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )
							assert dataset.num_rows == 4
							assert dataset.num_columns == 3
							assert dataset.column_names == ["col_1", "col_2", "col_3"]
							for feature, expected_dtype in expected_features.items():
														assert dataset.features[feature].dtype == expected_dtype



@pytest.mark.parametrize("""keep_in_memory"""     ,	[False, True]  )
def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    Optional[int]:
							_SCREAMING_SNAKE_CASE		:						List[str]			  =    tmp_path / """cache"""
							_SCREAMING_SNAKE_CASE		:						Union[str, Any]			  =    {"""col_1""": """string""", """col_2""": """int64""", """col_3""": """float64"""}
							with assert_arrow_memory_increases() if keep_in_memory else assert_arrow_memory_doesnt_increase():
														_SCREAMING_SNAKE_CASE		:						List[Any]			  =    ParquetDatasetReader(__SCREAMING_SNAKE_CASE     ,	cache_dir=__SCREAMING_SNAKE_CASE     ,	keep_in_memory=__SCREAMING_SNAKE_CASE  ).read()
							_check_parquet_dataset(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )



@pytest.mark.parametrize(
    """features"""     ,	[
        None,
        {"""col_1""": """string""", """col_2""": """int64""", """col_3""": """float64"""},
        {"""col_1""": """string""", """col_2""": """string""", """col_3""": """string"""},
        {"""col_1""": """int32""", """col_2""": """int32""", """col_3""": """int32"""},
        {"""col_1""": """float32""", """col_2""": """float32""", """col_3""": """float32"""},
    ]     ,	)
def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    Tuple:
							_SCREAMING_SNAKE_CASE		:						int			  =    tmp_path / """cache"""
							_SCREAMING_SNAKE_CASE		:						str			  =    {"""col_1""": """string""", """col_2""": """int64""", """col_3""": """float64"""}
							_SCREAMING_SNAKE_CASE		:						List[Any]			  =    features.copy() if features else default_expected_features
							_SCREAMING_SNAKE_CASE		:						List[Any]			  =    (
							    Features({feature: Value(__SCREAMING_SNAKE_CASE  ) for feature, dtype in features.items()}  ) if features is not None else None
							)
							_SCREAMING_SNAKE_CASE		:						Optional[Any]			  =    ParquetDatasetReader(__SCREAMING_SNAKE_CASE     ,	features=__SCREAMING_SNAKE_CASE     ,	cache_dir=__SCREAMING_SNAKE_CASE  ).read()
							_check_parquet_dataset(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )



@pytest.mark.parametrize("""split"""     ,	[None, NamedSplit("""train"""  ), """train""", """test"""]  )
def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    Tuple:
							_SCREAMING_SNAKE_CASE		:						Tuple			  =    tmp_path / """cache"""
							_SCREAMING_SNAKE_CASE		:						int			  =    {"""col_1""": """string""", """col_2""": """int64""", """col_3""": """float64"""}
							_SCREAMING_SNAKE_CASE		:						Dict			  =    ParquetDatasetReader(__SCREAMING_SNAKE_CASE     ,	cache_dir=__SCREAMING_SNAKE_CASE     ,	split=__SCREAMING_SNAKE_CASE  ).read()
							_check_parquet_dataset(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )
							assert dataset.split == split if split else "train"



@pytest.mark.parametrize("""path_type"""     ,	[str, list]  )
def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    str:
							if issubclass(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  ):
														_SCREAMING_SNAKE_CASE		:						Any			  =    parquet_path
							elif issubclass(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  ):
														_SCREAMING_SNAKE_CASE		:						Union[str, Any]			  =    [parquet_path]
							_SCREAMING_SNAKE_CASE		:						Optional[Any]			  =    tmp_path / """cache"""
							_SCREAMING_SNAKE_CASE		:						Union[str, Any]			  =    {"""col_1""": """string""", """col_2""": """int64""", """col_3""": """float64"""}
							_SCREAMING_SNAKE_CASE		:						str			  =    ParquetDatasetReader(__SCREAMING_SNAKE_CASE     ,	cache_dir=__SCREAMING_SNAKE_CASE  ).read()
							_check_parquet_dataset(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )



def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE=("train",)  )->    Union[str, Any]:
							assert isinstance(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )
							for split in splits:
														_SCREAMING_SNAKE_CASE		:						int			  =    dataset_dict[split]
														assert dataset.num_rows == 4
														assert dataset.num_columns == 3
														assert dataset.column_names == ["col_1", "col_2", "col_3"]
														for feature, expected_dtype in expected_features.items():
																					assert dataset.features[feature].dtype == expected_dtype



@pytest.mark.parametrize("""keep_in_memory"""     ,	[False, True]  )
def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    Optional[int]:
							_SCREAMING_SNAKE_CASE		:						Dict			  =    tmp_path / """cache"""
							_SCREAMING_SNAKE_CASE		:						Optional[Any]			  =    {"""col_1""": """string""", """col_2""": """int64""", """col_3""": """float64"""}
							with assert_arrow_memory_increases() if keep_in_memory else assert_arrow_memory_doesnt_increase():
														_SCREAMING_SNAKE_CASE		:						Tuple			  =    ParquetDatasetReader(
														    {"""train""": parquet_path}     ,	cache_dir=__SCREAMING_SNAKE_CASE     ,	keep_in_memory=__SCREAMING_SNAKE_CASE  ).read()
							_check_parquet_datasetdict(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )



@pytest.mark.parametrize(
    """features"""     ,	[
        None,
        {"""col_1""": """string""", """col_2""": """int64""", """col_3""": """float64"""},
        {"""col_1""": """string""", """col_2""": """string""", """col_3""": """string"""},
        {"""col_1""": """int32""", """col_2""": """int32""", """col_3""": """int32"""},
        {"""col_1""": """float32""", """col_2""": """float32""", """col_3""": """float32"""},
    ]     ,	)
def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    Dict:
							_SCREAMING_SNAKE_CASE		:						Optional[int]			  =    tmp_path / """cache"""
							_SCREAMING_SNAKE_CASE		:						Optional[Any]			  =    {"""col_1""": """string""", """col_2""": """int64""", """col_3""": """float64"""}
							_SCREAMING_SNAKE_CASE		:						List[str]			  =    features.copy() if features else default_expected_features
							_SCREAMING_SNAKE_CASE		:						str			  =    (
							    Features({feature: Value(__SCREAMING_SNAKE_CASE  ) for feature, dtype in features.items()}  ) if features is not None else None
							)
							_SCREAMING_SNAKE_CASE		:						int			  =    ParquetDatasetReader({"""train""": parquet_path}     ,	features=__SCREAMING_SNAKE_CASE     ,	cache_dir=__SCREAMING_SNAKE_CASE  ).read()
							_check_parquet_datasetdict(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )



@pytest.mark.parametrize("""split"""     ,	[None, NamedSplit("""train"""  ), """train""", """test"""]  )
def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    Dict:
							if split:
														_SCREAMING_SNAKE_CASE		:						Union[str, Any]			  =    {split: parquet_path}
							else:
														_SCREAMING_SNAKE_CASE		:						Optional[int]			  =    """train"""
														_SCREAMING_SNAKE_CASE		:						Any			  =    {"""train""": parquet_path, """test""": parquet_path}
							_SCREAMING_SNAKE_CASE		:						List[str]			  =    tmp_path / """cache"""
							_SCREAMING_SNAKE_CASE		:						List[str]			  =    {"""col_1""": """string""", """col_2""": """int64""", """col_3""": """float64"""}
							_SCREAMING_SNAKE_CASE		:						Union[str, Any]			  =    ParquetDatasetReader(__SCREAMING_SNAKE_CASE     ,	cache_dir=__SCREAMING_SNAKE_CASE  ).read()
							_check_parquet_datasetdict(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE     ,	splits=list(path.keys()  )  )
							assert all(dataset[split].split == split for split in path.keys()  )



def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    List[Any]:
							_SCREAMING_SNAKE_CASE		:						List[str]			  =    ParquetDatasetWriter(__SCREAMING_SNAKE_CASE     ,	tmp_path / """foo.parquet"""  )
							assert writer.write() > 0
							_SCREAMING_SNAKE_CASE		:						Tuple			  =    pq.ParquetFile(tmp_path / """foo.parquet"""  )
							_SCREAMING_SNAKE_CASE		:						str			  =    pf.read()
							assert dataset.data.table == output_table



def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    Union[str, Any]:
							_SCREAMING_SNAKE_CASE		:						Dict			  =    str(shared_datadir / """test_image_rgb.jpg"""  )
							_SCREAMING_SNAKE_CASE		:						Optional[Any]			  =    {"""image""": [image_path]}
							_SCREAMING_SNAKE_CASE		:						Optional[Any]			  =    Features({"""image""": Image()}  )
							_SCREAMING_SNAKE_CASE		:						Optional[int]			  =    Dataset.from_dict(__SCREAMING_SNAKE_CASE     ,	features=__SCREAMING_SNAKE_CASE  )
							_SCREAMING_SNAKE_CASE		:						Tuple			  =    ParquetDatasetWriter(__SCREAMING_SNAKE_CASE     ,	tmp_path / """foo.parquet"""  )
							assert writer.write() > 0

							_SCREAMING_SNAKE_CASE		:						List[str]			  =    Dataset.from_parquet(str(tmp_path / """foo.parquet"""  )  )
							assert dataset.features == reloaded_dataset.features

							_SCREAMING_SNAKE_CASE		:						List[Any]			  =    ParquetDatasetReader(str(tmp_path / """foo.parquet"""  )     ,	streaming=__SCREAMING_SNAKE_CASE  ).read()
							assert dataset.features == reloaded_iterable_dataset.features



@pytest.mark.parametrize(
    """feature, expected"""     ,	[
        (Features({"""foo""": Value("""int32"""  )}  ), None),
        (Features({"""image""": Image(), """foo""": Value("""int32"""  )}  ), config.PARQUET_ROW_GROUP_SIZE_FOR_IMAGE_DATASETS),
        (Features({"""nested""": Sequence(Audio()  )}  ), config.PARQUET_ROW_GROUP_SIZE_FOR_AUDIO_DATASETS),
    ]     ,	)
def      lowerCamelCase_(__SCREAMING_SNAKE_CASE     ,	__SCREAMING_SNAKE_CASE  )->    int:
							assert get_writer_batch_size(__SCREAMING_SNAKE_CASE  ) == expected


