




import argparse
from pathlib import Path
from typing import Dict, OrderedDict, Tuple

import torch
from audiocraft.models import MusicGen

from transformers import (
    AutoFeatureExtractor,
    AutoTokenizer,
    EncodecModel,
    MusicgenDecoderConfig,
    MusicgenForConditionalGeneration,
    MusicgenProcessor,
    TaEncoderModel,
)
from transformers.models.musicgen.modeling_musicgen import MusicgenForCausalLM
from transformers.utils import logging


logging.set_verbosity_info()
UpperCAmelCase_						=logging.get_logger(__name__)


UpperCAmelCase_						=["""model.decoder.embed_positions.weights"""]



def       UpperCAmelCase       (     _snake_case							):
							if "emb" in name:
														lowerCAmelCase   =		name.replace('''emb'''						, '''model.decoder.embed_tokens'''							)
							if "transformer" in name:
														lowerCAmelCase   =		name.replace('''transformer'''						, '''model.decoder'''							)
							if "cross_attention" in name:
														lowerCAmelCase   =		name.replace('''cross_attention'''						, '''encoder_attn'''							)
							if "linear1" in name:
														lowerCAmelCase   =		name.replace('''linear1'''						, '''fc1'''							)
							if "linear2" in name:
														lowerCAmelCase   =		name.replace('''linear2'''						, '''fc2'''							)
							if "norm1" in name:
														lowerCAmelCase   =		name.replace('''norm1'''						, '''self_attn_layer_norm'''							)
							if "norm_cross" in name:
														lowerCAmelCase   =		name.replace('''norm_cross'''						, '''encoder_attn_layer_norm'''							)
							if "norm2" in name:
														lowerCAmelCase   =		name.replace('''norm2'''						, '''final_layer_norm'''							)
							if "out_norm" in name:
														lowerCAmelCase   =		name.replace('''out_norm'''						, '''model.decoder.layer_norm'''							)
							if "linears" in name:
														lowerCAmelCase   =		name.replace('''linears'''						, '''lm_heads'''							)
							if "condition_provider.conditioners.description.output_proj" in name:
														lowerCAmelCase   =		name.replace('''condition_provider.conditioners.description.output_proj'''						, '''enc_to_dec_proj'''							)
							return name



def       UpperCAmelCase       (     _snake_case						, _snake_case							):
							lowerCAmelCase   =		list(state_dict.keys()							)
							lowerCAmelCase   =		{}
							for key in keys:
														lowerCAmelCase   =		state_dict.pop(_snake_case							)
														lowerCAmelCase   =		rename_keys(_snake_case							)
														if "in_proj_weight" in key:
																					# split fused qkv proj
																					lowerCAmelCase   =		val[:hidden_size, :]
																					lowerCAmelCase   =		val[hidden_size : 2 * hidden_size, :]
																					lowerCAmelCase   =		val[-hidden_size:, :]
														elif "enc_to_dec_proj" in key:
																					lowerCAmelCase   =		val
														else:
																					lowerCAmelCase   =		val
							return state_dict, enc_dec_proj_state_dict



def       UpperCAmelCase       (     _snake_case							):
							if checkpoint == "small":
														# default config values
														lowerCAmelCase   =		1024
														lowerCAmelCase   =		24
														lowerCAmelCase   =		16
							elif checkpoint == "medium":
														lowerCAmelCase   =		1536
														lowerCAmelCase   =		48
														lowerCAmelCase   =		24
							elif checkpoint == "large":
														lowerCAmelCase   =		2048
														lowerCAmelCase   =		48
														lowerCAmelCase   =		32
							else:
														raise ValueError(F"""Checkpoint should be one of `['small', 'medium', 'large']`, got {checkpoint}."""							)
							lowerCAmelCase   =		MusicgenDecoderConfig(
							    hidden_size=_snake_case						, ffn_dim=hidden_size * 4						, num_hidden_layers=_snake_case						, num_attention_heads=_snake_case						, )
							return config



@torch.no_grad()
def       UpperCAmelCase       (     _snake_case						, _snake_case=None						, _snake_case=None						, _snake_case="cpu"							):
							lowerCAmelCase   =		MusicGen.get_pretrained(_snake_case						, device=_snake_case							)
							lowerCAmelCase   =		decoder_config_from_checkpoint(_snake_case							)

							lowerCAmelCase   =		fairseq_model.lm.state_dict()
							lowerCAmelCase   ,     lowerCAmelCase   =		rename_state_dict(
							    _snake_case						, hidden_size=decoder_config.hidden_size							)

							lowerCAmelCase   =		TaEncoderModel.from_pretrained('''t5-base'''							)
							lowerCAmelCase   =		EncodecModel.from_pretrained('''facebook/encodec_32khz'''							)
							lowerCAmelCase   =		MusicgenForCausalLM(_snake_case							).eval()

							# load all decoder weights - expect that we'll be missing embeddings and enc-dec projection
							lowerCAmelCase   ,     lowerCAmelCase   =		decoder.load_state_dict(_snake_case						, strict=_snake_case							)

							for key in missing_keys.copy():
														if key.startswith(('''text_encoder''', '''audio_encoder''')							) or key in EXPECTED_MISSING_KEYS:
																					missing_keys.remove(_snake_case							)

							if len(_snake_case							) > 0:
														raise ValueError(F"""Missing key(s) in state_dict: {missing_keys}"""							)

							if len(_snake_case							) > 0:
														raise ValueError(F"""Unexpected key(s) in state_dict: {unexpected_keys}"""							)

							# init the composite model
							lowerCAmelCase   =		MusicgenForConditionalGeneration(text_encoder=_snake_case						, audio_encoder=_snake_case						, decoder=_snake_case							)

							# load the pre-trained enc-dec projection (from the decoder state dict)
							model.enc_to_dec_proj.load_state_dict(_snake_case							)

							# check we can do a forward pass
							lowerCAmelCase   =		torch.arange(0						, 8						, dtype=torch.long							).reshape(2						, -1							)
							lowerCAmelCase   =		input_ids.reshape(2 * 4						, -1							)

							with torch.no_grad():
														lowerCAmelCase   =		model(input_ids=_snake_case						, decoder_input_ids=_snake_case							).logits

							if logits.shape != (8, 1, 2048):
														raise ValueError('''Incorrect shape for logits'''							)

							# now construct the processor
							lowerCAmelCase   =		AutoTokenizer.from_pretrained('''t5-base'''							)
							lowerCAmelCase   =		AutoFeatureExtractor.from_pretrained('''facebook/encodec_32khz'''						, padding_side='''left'''							)

							lowerCAmelCase   =		MusicgenProcessor(feature_extractor=_snake_case						, tokenizer=_snake_case							)

							# set the appropriate bos/pad token ids
							lowerCAmelCase   =		2048
							lowerCAmelCase   =		2048

							# set other default generation config params
							lowerCAmelCase   =		int(30 * audio_encoder.config.frame_rate							)
							lowerCAmelCase   =		True
							lowerCAmelCase   =		3.0

							if pytorch_dump_folder is not None:
														Path(_snake_case							).mkdir(exist_ok=_snake_case							)
														logger.info(F"""Saving model {checkpoint} to {pytorch_dump_folder}"""							)
														model.save_pretrained(_snake_case							)
														processor.save_pretrained(_snake_case							)

							if repo_id:
														logger.info(F"""Pushing model {checkpoint} to {repo_id}"""							)
														model.push_to_hub(_snake_case							)
														processor.push_to_hub(_snake_case							)


if __name__ == "__main__":
		UpperCAmelCase_						=argparse.ArgumentParser()
		# Required parameters
		parser.add_argument(
		    """--checkpoint""",
		    default="""small""",
		    type=str,
		    help="""Checkpoint size of the MusicGen model you'd like to convert. Can be one of: `['small', 'medium', 'large']`.""",
		)
		parser.add_argument(
		    """--pytorch_dump_folder""",
		    required=True,
		    default=None,
		    type=str,
		    help="""Path to the output PyTorch model directory.""",
		)
		parser.add_argument(
		    """--push_to_hub""", default=None, type=str, help="""Where to upload the converted model on the 🤗 hub."""
		)
		parser.add_argument(
		    """--device""", default="""cpu""", type=str, help="""Torch device to run the conversion, either cpu or cuda."""
		)

		UpperCAmelCase_						=parser.parse_args()
		convert_musicgen_checkpoint(args.checkpoint, args.pytorch_dump_folder, args.push_to_hub)









import os
import pickle
import unittest

from transformers import AutoTokenizer
from transformers.models.bert.tokenization_bert import BertTokenizer
from transformers.models.bert_japanese.tokenization_bert_japanese import (
    VOCAB_FILES_NAMES,
    BertJapaneseTokenizer,
    CharacterTokenizer,
    JumanppTokenizer,
    MecabTokenizer,
    SudachiTokenizer,
    WordpieceTokenizer,
)
from transformers.testing_utils import custom_tokenizers, require_jumanpp, require_sudachi

from ...test_tokenization_common import TokenizerTesterMixin


@custom_tokenizers
class 							__UpperCamelCase			( __UpperCAmelCase     ,					unittest.TestCase  ):





				'''simple docstring'''



				__a							: Any										=BertJapaneseTokenizer
				__a							: Optional[int]										=False
				__a							: int										=True



				def       __snake_case		( self						):
											super().setUp()

											lowerCAmelCase   =		[
											    '''[UNK]''',
											    '''[CLS]''',
											    '''[SEP]''',
											    '''こんにちは''',
											    '''こん''',
											    '''にちは''',
											    '''ばんは''',
											    '''##こん''',
											    '''##にちは''',
											    '''##ばんは''',
											    '''世界''',
											    '''##世界''',
											    '''、''',
											    '''##、''',
											    '''。''',
											    '''##。''',
											]

											lowerCAmelCase   =		os.path.join(self.tmpdirname       , VOCAB_FILES_NAMES['''vocab_file''']						)
											with open(self.vocab_file       , '''w'''       , encoding='''utf-8'''						) as vocab_writer:
																		vocab_writer.write(''''''.join([x + '''\n''' for x in vocab_tokens]						)						)



				def       __snake_case		( self       , UpperCAmelCase_						):
											lowerCAmelCase   =		'''こんにちは、世界。 \nこんばんは、世界。'''
											lowerCAmelCase   =		'''こんにちは 、 世界 。 こんばんは 、 世界 。'''
											return input_text, output_text



				def       __snake_case		( self       , UpperCAmelCase_						):
											lowerCAmelCase   ,     lowerCAmelCase   =		self.get_input_output_texts(UpperCAmelCase_						)
											lowerCAmelCase   =		tokenizer.encode(UpperCAmelCase_       , add_special_tokens=UpperCAmelCase_						)
											lowerCAmelCase   =		tokenizer.decode(UpperCAmelCase_       , clean_up_tokenization_spaces=UpperCAmelCase_						)
											return text, ids



				def       __snake_case		( self						):
											pass  # TODO add if relevant



				def       __snake_case		( self						):
											pass  # TODO add if relevant



				def       __snake_case		( self						):
											pass  # TODO add if relevant



				def       __snake_case		( self						):
											lowerCAmelCase   =		self.tokenizer_class(self.vocab_file						)

											lowerCAmelCase   =		tokenizer.tokenize('''こんにちは、世界。\nこんばんは、世界。'''						)
											self.assertListEqual(UpperCAmelCase_       , ['''こんにちは''', '''、''', '''世界''', '''。''', '''こん''', '''##ばんは''', '''、''', '''世界''', '''。''']						)
											self.assertListEqual(tokenizer.convert_tokens_to_ids(UpperCAmelCase_						)       , [3, 12, 10, 14, 4, 9, 12, 10, 14]						)



				def       __snake_case		( self						):
											lowerCAmelCase   =		self.tokenizer_class(self.vocab_file       , word_tokenizer_type='''mecab'''						)
											self.assertIsNotNone(UpperCAmelCase_						)

											lowerCAmelCase   =		'''こんにちは、世界。\nこんばんは、世界。'''
											lowerCAmelCase   =		tokenizer.tokenize(UpperCAmelCase_						)
											self.assertListEqual(UpperCAmelCase_       , ['''こんにちは''', '''、''', '''世界''', '''。''', '''こん''', '''##ばんは''', '''、''', '''世界''', '''。''']						)
											self.assertListEqual(tokenizer.convert_tokens_to_ids(UpperCAmelCase_						)       , [3, 12, 10, 14, 4, 9, 12, 10, 14]						)

											lowerCAmelCase   =		os.path.join(self.tmpdirname       , '''tokenizer.bin'''						)
											with open(UpperCAmelCase_       , '''wb'''						) as handle:
																		pickle.dump(UpperCAmelCase_       , UpperCAmelCase_						)

											with open(UpperCAmelCase_       , '''rb'''						) as handle:
																		lowerCAmelCase   =		pickle.load(UpperCAmelCase_						)

											lowerCAmelCase   =		tokenizer_new.tokenize(UpperCAmelCase_						)

											self.assertListEqual(UpperCAmelCase_       , UpperCAmelCase_						)



				def       __snake_case		( self						):
											lowerCAmelCase   =		MecabTokenizer(mecab_dic='''ipadic'''						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''アップルストア''', '''で''', '''iPhone''', '''8''', '''が''', '''発売''', '''さ''', '''れ''', '''た''', '''。''']       , )



				def       __snake_case		( self						):
											try:
																		lowerCAmelCase   =		MecabTokenizer(mecab_dic='''unidic_lite'''						)
											except ModuleNotFoundError:
																		return

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''アップル''', '''ストア''', '''で''', '''iPhone''', '''8''', '''が''', '''発売''', '''さ''', '''れ''', '''た''', '''。''']       , )



				def       __snake_case		( self						):
											try:
																		lowerCAmelCase   =		MecabTokenizer(mecab_dic='''unidic'''						)
											except ModuleNotFoundError:
																		return

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''アップル''', '''ストア''', '''で''', '''iPhone''', '''8''', '''が''', '''発売''', '''さ''', '''れ''', '''た''', '''。''']       , )



				def       __snake_case		( self						):
											lowerCAmelCase   =		MecabTokenizer(do_lower_case=UpperCAmelCase_       , mecab_dic='''ipadic'''						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''アップルストア''', '''で''', '''iphone''', '''8''', '''が''', '''発売''', '''さ''', '''れ''', '''た''', '''。''']       , )



				def       __snake_case		( self						):
											try:
																		lowerCAmelCase   =		MecabTokenizer(
																		    do_lower_case=UpperCAmelCase_       , normalize_text=UpperCAmelCase_       , mecab_option='''-d /usr/local/lib/mecab/dic/jumandic'''						)
											except RuntimeError:
																		# if dict doesn't exist in the system, previous code raises this error.
																		return

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''ｱｯﾌﾟﾙストア''', '''で''', '''iPhone''', '''８''', '''が''', '''発売''', '''さ''', '''れた''', '''\u3000''', '''。''']       , )



				def       __snake_case		( self						):
											lowerCAmelCase   =		MecabTokenizer(normalize_text=UpperCAmelCase_       , mecab_dic='''ipadic'''						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''ｱｯﾌﾟﾙストア''', '''で''', '''iPhone''', '''８''', '''が''', '''発売''', '''さ''', '''れ''', '''た''', '''　''', '''。''']       , )



				@require_sudachi
				def       __snake_case		( self						):
											lowerCAmelCase   =		self.tokenizer_class(self.vocab_file       , word_tokenizer_type='''sudachi'''						)
											self.assertIsNotNone(UpperCAmelCase_						)

											lowerCAmelCase   =		'''こんにちは、世界。\nこんばんは、世界。'''
											lowerCAmelCase   =		tokenizer.tokenize(UpperCAmelCase_						)
											self.assertListEqual(UpperCAmelCase_       , ['''こんにちは''', '''、''', '''世界''', '''。''', '''こん''', '''##ばんは''', '''、''', '''世界''', '''。''']						)
											self.assertListEqual(tokenizer.convert_tokens_to_ids(UpperCAmelCase_						)       , [3, 12, 10, 14, 4, 9, 12, 10, 14]						)

											lowerCAmelCase   =		os.path.join(self.tmpdirname       , '''tokenizer.bin'''						)
											with open(UpperCAmelCase_       , '''wb'''						) as handle:
																		pickle.dump(UpperCAmelCase_       , UpperCAmelCase_						)

											with open(UpperCAmelCase_       , '''rb'''						) as handle:
																		lowerCAmelCase   =		pickle.load(UpperCAmelCase_						)

											lowerCAmelCase   =		tokenizer_new.tokenize(UpperCAmelCase_						)

											self.assertListEqual(UpperCAmelCase_       , UpperCAmelCase_						)



				@require_sudachi
				def       __snake_case		( self						):
											lowerCAmelCase   =		SudachiTokenizer(sudachi_dict_type='''core'''						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , [''' ''', '''\t''', '''アップル''', '''ストア''', '''で''', '''iPhone''', '''8''', ''' ''', '''が''', ''' ''', ''' ''', '''\n ''', '''発売''', '''さ''', '''れ''', '''た''', ''' ''', '''。''', ''' ''', ''' ''']       , )



				@require_sudachi
				def       __snake_case		( self						):
											lowerCAmelCase   =		SudachiTokenizer(sudachi_dict_type='''core'''       , sudachi_split_mode='''A'''						)

											self.assertListEqual(tokenizer.tokenize('''外国人参政権'''						)       , ['''外国''', '''人''', '''参政''', '''権''']						)



				@require_sudachi
				def       __snake_case		( self						):
											lowerCAmelCase   =		SudachiTokenizer(sudachi_dict_type='''core'''       , sudachi_split_mode='''B'''						)

											self.assertListEqual(tokenizer.tokenize('''外国人参政権'''						)       , ['''外国人''', '''参政権''']						)



				@require_sudachi
				def       __snake_case		( self						):
											lowerCAmelCase   =		SudachiTokenizer(sudachi_dict_type='''core'''       , sudachi_split_mode='''C'''						)

											self.assertListEqual(tokenizer.tokenize('''外国人参政権'''						)       , ['''外国人参政権''']						)



				@require_sudachi
				def       __snake_case		( self						):
											lowerCAmelCase   =		SudachiTokenizer(do_lower_case=UpperCAmelCase_       , sudachi_dict_type='''core'''						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , [''' ''', '''\t''', '''アップル''', '''ストア''', '''で''', '''iphone''', '''8''', ''' ''', '''が''', ''' ''', ''' ''', '''\n ''', '''発売''', '''さ''', '''れ''', '''た''', ''' ''', '''。''', ''' ''', ''' ''']       , )



				@require_sudachi
				def       __snake_case		( self						):
											lowerCAmelCase   =		SudachiTokenizer(normalize_text=UpperCAmelCase_       , sudachi_dict_type='''core'''						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , [''' ''', '''\t''', '''ｱｯﾌﾟﾙ''', '''ストア''', '''で''', '''iPhone''', '''８''', ''' ''', '''が''', ''' ''', ''' ''', '''\n ''', '''発売''', '''さ''', '''れ''', '''た''', '''\u3000''', '''。''', ''' ''', ''' ''']       , )



				@require_sudachi
				def       __snake_case		( self						):
											lowerCAmelCase   =		SudachiTokenizer(trim_whitespace=UpperCAmelCase_       , sudachi_dict_type='''core'''						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''アップル''', '''ストア''', '''で''', '''iPhone''', '''8''', '''が''', '''発売''', '''さ''', '''れ''', '''た''', '''。''']       , )



				@require_jumanpp
				def       __snake_case		( self						):
											lowerCAmelCase   =		self.tokenizer_class(self.vocab_file       , word_tokenizer_type='''jumanpp'''						)
											self.assertIsNotNone(UpperCAmelCase_						)

											lowerCAmelCase   =		'''こんにちは、世界。\nこんばんは、世界。'''
											lowerCAmelCase   =		tokenizer.tokenize(UpperCAmelCase_						)
											self.assertListEqual(UpperCAmelCase_       , ['''こんにちは''', '''、''', '''世界''', '''。''', '''こん''', '''##ばんは''', '''、''', '''世界''', '''。''']						)
											self.assertListEqual(tokenizer.convert_tokens_to_ids(UpperCAmelCase_						)       , [3, 12, 10, 14, 4, 9, 12, 10, 14]						)

											lowerCAmelCase   =		os.path.join(self.tmpdirname       , '''tokenizer.bin'''						)
											with open(UpperCAmelCase_       , '''wb'''						) as handle:
																		pickle.dump(UpperCAmelCase_       , UpperCAmelCase_						)

											with open(UpperCAmelCase_       , '''rb'''						) as handle:
																		lowerCAmelCase   =		pickle.load(UpperCAmelCase_						)

											lowerCAmelCase   =		tokenizer_new.tokenize(UpperCAmelCase_						)

											self.assertListEqual(UpperCAmelCase_       , UpperCAmelCase_						)



				@require_jumanpp
				def       __snake_case		( self						):
											lowerCAmelCase   =		JumanppTokenizer()

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''アップル''', '''ストア''', '''で''', '''iPhone''', '''8''', '''\u3000''', '''が''', '''\u3000''', '''\u3000''', '''\u3000''', '''発売''', '''さ''', '''れた''', '''\u3000''', '''。''']       , )



				@require_jumanpp
				def       __snake_case		( self						):
											lowerCAmelCase   =		JumanppTokenizer(do_lower_case=UpperCAmelCase_						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''アップル''', '''ストア''', '''で''', '''iphone''', '''8''', '''\u3000''', '''が''', '''\u3000''', '''\u3000''', '''\u3000''', '''発売''', '''さ''', '''れた''', '''\u3000''', '''。''']       , )



				@require_jumanpp
				def       __snake_case		( self						):
											lowerCAmelCase   =		JumanppTokenizer(normalize_text=UpperCAmelCase_						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''ｱ''', '''ｯ''', '''ﾌ''', '''ﾟ''', '''ﾙ''', '''ストア''', '''で''', '''iPhone''', '''８''', '''\u3000''', '''が''', '''\u3000''', '''\u3000''', '''\u3000''', '''発売''', '''さ''', '''れた''', '''\u3000''', '''。''']       , )



				@require_jumanpp
				def       __snake_case		( self						):
											lowerCAmelCase   =		JumanppTokenizer(trim_whitespace=UpperCAmelCase_						)

											self.assertListEqual(
											    tokenizer.tokenize(''' \tｱｯﾌﾟﾙストアでiPhone８ が  \n 発売された　。  '''						)       , ['''アップル''', '''ストア''', '''で''', '''iPhone''', '''8''', '''が''', '''発売''', '''さ''', '''れた''', '''。''']       , )



				@require_jumanpp
				def       __snake_case		( self						):
											lowerCAmelCase   =		JumanppTokenizer()

											self.assertListEqual(
											    tokenizer.tokenize('''ありがとうございますm(_ _)ｍ見つけるのが大変です。'''						)       , ['''ありがとう''', '''ございます''', '''m(_ _)m''', '''見つける''', '''の''', '''が''', '''大変です''', '''。''']       , )



				def       __snake_case		( self						):
											lowerCAmelCase   =		['''[UNK]''', '''[CLS]''', '''[SEP]''', '''こんにちは''', '''こん''', '''にちは''', '''ばんは''', '''##こん''', '''##にちは''', '''##ばんは''']

											lowerCAmelCase   =		{}
											for i, token in enumerate(UpperCAmelCase_						):
																		lowerCAmelCase   =		i
											lowerCAmelCase   =		WordpieceTokenizer(vocab=UpperCAmelCase_       , unk_token='''[UNK]'''						)

											self.assertListEqual(tokenizer.tokenize(''''''						)       , []						)

											self.assertListEqual(tokenizer.tokenize('''こんにちは'''						)       , ['''こんにちは''']						)

											self.assertListEqual(tokenizer.tokenize('''こんばんは'''						)       , ['''こん''', '''##ばんは''']						)

											self.assertListEqual(tokenizer.tokenize('''こんばんは こんばんにちは こんにちは'''						)       , ['''こん''', '''##ばんは''', '''[UNK]''', '''こんにちは''']						)



				def       __snake_case		( self						):
											lowerCAmelCase   =		BertJapaneseTokenizer.from_pretrained('''nlp-waseda/roberta-base-japanese-with-auto-jumanpp'''						)
											lowerCAmelCase   =		tokenizer.subword_tokenizer

											lowerCAmelCase   =		subword_tokenizer.tokenize('''国境 の 長い トンネル を 抜ける と 雪国 であった 。'''						)
											self.assertListEqual(UpperCAmelCase_       , ['''▁国境''', '''▁の''', '''▁長い''', '''▁トンネル''', '''▁を''', '''▁抜ける''', '''▁と''', '''▁雪''', '''国''', '''▁であった''', '''▁。''']						)

											lowerCAmelCase   =		subword_tokenizer.tokenize('''こんばんは こんばん にち は こんにちは'''						)
											self.assertListEqual(UpperCAmelCase_       , ['''▁こん''', '''ばん''', '''は''', '''▁こん''', '''ばん''', '''▁に''', '''ち''', '''▁は''', '''▁こんにちは''']						)






				def       __snake_case		( self						):
											lowerCAmelCase   =		self.tokenizer_class.from_pretrained('''cl-tohoku/bert-base-japanese'''						)

											lowerCAmelCase   =		tokenizer.encode('''ありがとう。'''       , add_special_tokens=UpperCAmelCase_						)
											lowerCAmelCase   =		tokenizer.encode('''どういたしまして。'''       , add_special_tokens=UpperCAmelCase_						)

											lowerCAmelCase   =		tokenizer.build_inputs_with_special_tokens(UpperCAmelCase_						)
											lowerCAmelCase   =		tokenizer.build_inputs_with_special_tokens(UpperCAmelCase_       , UpperCAmelCase_						)

											# 2 is for "[CLS]", 3 is for "[SEP]"
											assert encoded_sentence == [2] + text + [3]
											assert encoded_pair == [2] + text + [3] + text_a + [3]




@custom_tokenizers
class 							__UpperCamelCase			( __UpperCAmelCase     ,					unittest.TestCase  ):





				'''simple docstring'''



				__a							: Union[str, Any]										=BertJapaneseTokenizer
				__a							: Optional[int]										=False



				def       __snake_case		( self						):
											super().setUp()

											lowerCAmelCase   =		['''[UNK]''', '''[CLS]''', '''[SEP]''', '''こ''', '''ん''', '''に''', '''ち''', '''は''', '''ば''', '''世''', '''界''', '''、''', '''。''']

											lowerCAmelCase   =		os.path.join(self.tmpdirname       , VOCAB_FILES_NAMES['''vocab_file''']						)
											with open(self.vocab_file       , '''w'''       , encoding='''utf-8'''						) as vocab_writer:
																		vocab_writer.write(''''''.join([x + '''\n''' for x in vocab_tokens]						)						)



				def       __snake_case		( self       , **UpperCAmelCase_						):
											return BertJapaneseTokenizer.from_pretrained(self.tmpdirname       , subword_tokenizer_type='''character'''       , **UpperCAmelCase_						)



				def       __snake_case		( self       , UpperCAmelCase_						):
											lowerCAmelCase   =		'''こんにちは、世界。 \nこんばんは、世界。'''
											lowerCAmelCase   =		'''こ ん に ち は 、 世 界 。 こ ん ば ん は 、 世 界 。'''
											return input_text, output_text



				def       __snake_case		( self						):
											pass  # TODO add if relevant



				def       __snake_case		( self						):
											pass  # TODO add if relevant



				def       __snake_case		( self						):
											pass  # TODO add if relevant



				def       __snake_case		( self						):
											lowerCAmelCase   =		self.tokenizer_class(self.vocab_file       , subword_tokenizer_type='''character'''						)

											lowerCAmelCase   =		tokenizer.tokenize('''こんにちは、世界。 \nこんばんは、世界。'''						)
											self.assertListEqual(
											    UpperCAmelCase_       , ['''こ''', '''ん''', '''に''', '''ち''', '''は''', '''、''', '''世''', '''界''', '''。''', '''こ''', '''ん''', '''ば''', '''ん''', '''は''', '''、''', '''世''', '''界''', '''。''']						)
											self.assertListEqual(
											    tokenizer.convert_tokens_to_ids(UpperCAmelCase_						)       , [3, 4, 5, 6, 7, 11, 9, 10, 12, 3, 4, 8, 4, 7, 11, 9, 10, 12]						)



				def       __snake_case		( self						):
											lowerCAmelCase   =		['''[UNK]''', '''[CLS]''', '''[SEP]''', '''こ''', '''ん''', '''に''', '''ち''', '''は''', '''ば''', '''世''', '''界''', '''、''', '''。''']

											lowerCAmelCase   =		{}
											for i, token in enumerate(UpperCAmelCase_						):
																		lowerCAmelCase   =		i
											lowerCAmelCase   =		CharacterTokenizer(vocab=UpperCAmelCase_       , unk_token='''[UNK]'''						)

											self.assertListEqual(tokenizer.tokenize(''''''						)       , []						)

											self.assertListEqual(tokenizer.tokenize('''こんにちは'''						)       , ['''こ''', '''ん''', '''に''', '''ち''', '''は''']						)

											self.assertListEqual(tokenizer.tokenize('''こんにちほ'''						)       , ['''こ''', '''ん''', '''に''', '''ち''', '''[UNK]''']						)






				def       __snake_case		( self						):
											lowerCAmelCase   =		self.tokenizer_class.from_pretrained('''cl-tohoku/bert-base-japanese-char'''						)

											lowerCAmelCase   =		tokenizer.encode('''ありがとう。'''       , add_special_tokens=UpperCAmelCase_						)
											lowerCAmelCase   =		tokenizer.encode('''どういたしまして。'''       , add_special_tokens=UpperCAmelCase_						)

											lowerCAmelCase   =		tokenizer.build_inputs_with_special_tokens(UpperCAmelCase_						)
											lowerCAmelCase   =		tokenizer.build_inputs_with_special_tokens(UpperCAmelCase_       , UpperCAmelCase_						)

											# 2 is for "[CLS]", 3 is for "[SEP]"
											assert encoded_sentence == [2] + text + [3]
											assert encoded_pair == [2] + text + [3] + text_a + [3]




@custom_tokenizers
class 							__UpperCamelCase			( unittest.TestCase  ):





				'''simple docstring'''



				def       __snake_case		( self						):
											lowerCAmelCase   =		'''cl-tohoku/bert-base-japanese'''
											lowerCAmelCase   =		AutoTokenizer.from_pretrained(UpperCAmelCase_						)
											self.assertIsInstance(UpperCAmelCase_       , UpperCAmelCase_						)




class 							__UpperCamelCase			( unittest.TestCase  ):





				'''simple docstring'''



				def       __snake_case		( self						):
											lowerCAmelCase   =		'''cl-tohoku/bert-base-japanese'''
											with self.assertLogs('''transformers'''       , level='''WARNING'''						) as cm:
																		BertTokenizer.from_pretrained(UpperCAmelCase_						)
																		self.assertTrue(
																		    cm.records[0].message.startswith(
																		        '''The tokenizer class you load from this checkpoint is not the same type as the class this function'''
																		        ''' is called from.'''						)						)
											lowerCAmelCase   =		'''bert-base-cased'''
											with self.assertLogs('''transformers'''       , level='''WARNING'''						) as cm:
																		BertJapaneseTokenizer.from_pretrained(UpperCAmelCase_						)
																		self.assertTrue(
																		    cm.records[0].message.startswith(
																		        '''The tokenizer class you load from this checkpoint is not the same type as the class this function'''
																		        ''' is called from.'''						)						)



