



import json
import logging
import os
import sys
from time import time
from unittest.mock import patch

from transformers.testing_utils import TestCasePlus, require_torch_tpu


logging.basicConfig(level=logging.DEBUG)

SCREAMING_SNAKE_CASE     				=						logging.getLogger()







def snake_case__  (			__SCREAMING_SNAKE_CASE )		->							Optional[int]:
		UpperCAmelCase_						=				{}
		UpperCAmelCase_						=				os.path.join(_lowercase			,     "all_results.json" )
		if os.path.exists(_lowercase ):
				with open(_lowercase			,     "r" ) as f:
						UpperCAmelCase_						=				json.load(_lowercase )
		else:
				raise ValueError(f'''can\'t find {path}''' )
		return results


SCREAMING_SNAKE_CASE     				=						logging.StreamHandler(sys.stdout)
logger.addHandler(stream_handler)



@require_torch_tpu
class 	lowerCamelCase      (						lowercase__  ):







					'''simple docstring'''







					def    A__		(			self     ):
							import xla_spawn

							UpperCAmelCase_						=				self.get_auto_remove_tmp_dir()
							UpperCAmelCase_						=				f'''
            ./examples/pytorch/text-classification/run_glue.py
            --num_cores=8
            ./examples/pytorch/text-classification/run_glue.py
            --model_name_or_path distilbert-base-uncased
            --output_dir {tmp_dir}
            --overwrite_output_dir
            --train_file ./tests/fixtures/tests_samples/MRPC/train.csv
            --validation_file ./tests/fixtures/tests_samples/MRPC/dev.csv
            --do_train
            --do_eval
            --debug tpu_metrics_debug
            --per_device_train_batch_size=2
            --per_device_eval_batch_size=1
            --learning_rate=1e-4
            --max_steps=10
            --warmup_steps=2
            --seed=42
            --max_seq_length=128
            '''.split()

							with patch.object(UpperCamelCase_    ,	"argv"    ,	UpperCamelCase_     ):
									UpperCAmelCase_						=				time()
									xla_spawn.main()
									UpperCAmelCase_						=				time()

									UpperCAmelCase_						=				get_results(UpperCamelCase_     )
									self.assertGreaterEqual(result["eval_accuracy"]    ,	0.75     )

									# Assert that the script takes less than 500 seconds to make sure it doesn't hang.
									self.assertLess(end - start    ,	500     )



					def    A__		(			self     ):
							import xla_spawn

							UpperCAmelCase_						=				'\n            ./tests/test_trainer_tpu.py\n            --num_cores=8\n            ./tests/test_trainer_tpu.py\n            '.split()
							with patch.object(UpperCamelCase_    ,	"argv"    ,	UpperCamelCase_     ):
									xla_spawn.main()










import json
from typing import List, Optional, Tuple

from tokenizers import normalizers

from ...tokenization_utils_fast import PreTrainedTokenizerFast
from ...utils import logging
from .tokenization_distilbert import DistilBertTokenizer


SCREAMING_SNAKE_CASE     				=						logging.get_logger(__name__)

SCREAMING_SNAKE_CASE     				=						{"vocab_file": "vocab.txt", "tokenizer_file": "tokenizer.json"}

SCREAMING_SNAKE_CASE     				=						{
    "vocab_file": {
        "distilbert-base-uncased": "https://huggingface.co/distilbert-base-uncased/resolve/main/vocab.txt",
        "distilbert-base-uncased-distilled-squad": (
            "https://huggingface.co/distilbert-base-uncased-distilled-squad/resolve/main/vocab.txt"
        ),
        "distilbert-base-cased": "https://huggingface.co/distilbert-base-cased/resolve/main/vocab.txt",
        "distilbert-base-cased-distilled-squad": (
            "https://huggingface.co/distilbert-base-cased-distilled-squad/resolve/main/vocab.txt"
        ),
        "distilbert-base-german-cased": "https://huggingface.co/distilbert-base-german-cased/resolve/main/vocab.txt",
        "distilbert-base-multilingual-cased": (
            "https://huggingface.co/distilbert-base-multilingual-cased/resolve/main/vocab.txt"
        ),
    },
    "tokenizer_file": {
        "distilbert-base-uncased": "https://huggingface.co/distilbert-base-uncased/resolve/main/tokenizer.json",
        "distilbert-base-uncased-distilled-squad": (
            "https://huggingface.co/distilbert-base-uncased-distilled-squad/resolve/main/tokenizer.json"
        ),
        "distilbert-base-cased": "https://huggingface.co/distilbert-base-cased/resolve/main/tokenizer.json",
        "distilbert-base-cased-distilled-squad": (
            "https://huggingface.co/distilbert-base-cased-distilled-squad/resolve/main/tokenizer.json"
        ),
        "distilbert-base-german-cased": (
            "https://huggingface.co/distilbert-base-german-cased/resolve/main/tokenizer.json"
        ),
        "distilbert-base-multilingual-cased": (
            "https://huggingface.co/distilbert-base-multilingual-cased/resolve/main/tokenizer.json"
        ),
    },
}

SCREAMING_SNAKE_CASE     				=						{
    "distilbert-base-uncased": 512,
    "distilbert-base-uncased-distilled-squad": 512,
    "distilbert-base-cased": 512,
    "distilbert-base-cased-distilled-squad": 512,
    "distilbert-base-german-cased": 512,
    "distilbert-base-multilingual-cased": 512,
}


SCREAMING_SNAKE_CASE     				=						{
    "distilbert-base-uncased": {"do_lower_case": True},
    "distilbert-base-uncased-distilled-squad": {"do_lower_case": True},
    "distilbert-base-cased": {"do_lower_case": False},
    "distilbert-base-cased-distilled-squad": {"do_lower_case": False},
    "distilbert-base-german-cased": {"do_lower_case": False},
    "distilbert-base-multilingual-cased": {"do_lower_case": False},
}



class 	lowerCamelCase      (						lowercase__  ):







					'''simple docstring'''







					lowerCAmelCase_				:    Any									=  VOCAB_FILES_NAMES
					lowerCAmelCase_				:    List[str]									=  PRETRAINED_VOCAB_FILES_MAP
					lowerCAmelCase_				:    str									=  PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
					lowerCAmelCase_				:    Union[str, Any]									=  PRETRAINED_INIT_CONFIGURATION
					lowerCAmelCase_				:    int									=  ['input_ids', 'attention_mask']
					lowerCAmelCase_				:    str									=  DistilBertTokenizer



					def __init__(			self    ,	lowerCAmelCase=None    ,	lowerCAmelCase=None    ,	lowerCAmelCase=True    ,	lowerCAmelCase="[UNK]"    ,	lowerCAmelCase="[SEP]"    ,	lowerCAmelCase="[PAD]"    ,	lowerCAmelCase="[CLS]"    ,	lowerCAmelCase="[MASK]"    ,	lowerCAmelCase=True    ,	lowerCAmelCase=None    ,	**lowerCAmelCase    ,	):
							super().__init__(
							    lowerCAmelCase    ,	tokenizer_file=lowerCAmelCase    ,	do_lower_case=lowerCAmelCase    ,	unk_token=lowerCAmelCase    ,	sep_token=lowerCAmelCase    ,	pad_token=lowerCAmelCase    ,	cls_token=lowerCAmelCase    ,	mask_token=lowerCAmelCase    ,	tokenize_chinese_chars=lowerCAmelCase    ,	strip_accents=lowerCAmelCase    ,	**lowerCAmelCase    ,	)

							UpperCAmelCase_						=				json.loads(self.backend_tokenizer.normalizer.__getstate__()     )
							if (
							    normalizer_state.get("lowercase"    ,	lowerCAmelCase     ) != do_lower_case
							    or normalizer_state.get("strip_accents"    ,	lowerCAmelCase     ) != strip_accents
							    or normalizer_state.get("handle_chinese_chars"    ,	lowerCAmelCase     ) != tokenize_chinese_chars
							):
									UpperCAmelCase_						=				getattr(lowerCAmelCase    ,	normalizer_state.pop("type"     )     )
									UpperCAmelCase_						=				do_lower_case
									UpperCAmelCase_						=				strip_accents
									UpperCAmelCase_						=				tokenize_chinese_chars
									UpperCAmelCase_						=				normalizer_class(**lowerCAmelCase     )

							UpperCAmelCase_						=				do_lower_case



					def    A__		(			self    ,	lowerCAmelCase    ,	lowerCAmelCase=None     ):
							UpperCAmelCase_						=				[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    A__		(			self    ,	lowerCAmelCase    ,	lowerCAmelCase = None     ):
							UpperCAmelCase_						=				[self.sep_token_id]
							UpperCAmelCase_						=				[self.cls_token_id]
							if token_ids_a is None:
									return len(cls + token_ids_a + sep     ) * [0]
							return len(cls + token_ids_a + sep     ) * [0] + len(token_ids_a + sep     ) * [1]



					def    A__		(			self    ,	lowerCAmelCase    ,	lowerCAmelCase = None     ):
							UpperCAmelCase_						=				self._tokenizer.model.save(lowerCAmelCase    ,	name=lowerCAmelCase     )
							return tuple(lowerCAmelCase     )

