
# this script builds a small sample spm file tests/fixtures/test_sentencepiece_no_bos.model, with features needed by pegasus

# 1. pip install sentencepiece
#
# 2. wget https://raw.githubusercontent.com/google/sentencepiece/master/data/botchan.txt

# 3. build
import sentencepiece as spm

# pegasus:
# 1. no bos
# 2. eos_id is 1
# 3. unk_id is 2
# build a sample spm file accordingly
spm.SentencePieceTrainer.train(
    """--input=botchan.txt --model_prefix=test_sentencepiece_no_bos --bos_id=-1 --unk_id=2  --eos_id=1  --vocab_size=1000"""
)

# 4. now update the fixture
# mv test_sentencepiece_no_bos.model ../../tests/fixtures/





import unittest
from pathlib import Path
from tempfile import TemporaryDirectory
from unittest.mock import Mock, patch

import diffusers.utils.hub_utils





class 						lowercase       (  unittest.TestCase					):
					@patch('''diffusers.utils.hub_utils.get_full_repo_name'''    )
					def lowercase_			(		self ,	SCREAMING_SNAKE_CASE__    ):
								"""simple docstring"""







								lowerCAmelCase__ :    str				      =			'''full_repo_name'''
								with TemporaryDirectory() as tmpdir:
											# Dummy args values
											lowerCAmelCase__ :    Optional[int]				      =			Mock()
											lowerCAmelCase__ :    Optional[int]				      =			tmpdir
											lowerCAmelCase__ :    Optional[Any]				      =			0
											lowerCAmelCase__ :    Optional[int]				      =			'''hub_token'''
											lowerCAmelCase__ :    Tuple				      =			'''dataset_name'''
											lowerCAmelCase__ :    int				      =			0.01
											lowerCAmelCase__ :    Optional[int]				      =			100000
											lowerCAmelCase__ :    Any				      =			10000
											lowerCAmelCase__ :    Union[str, Any]				      =			0.01
											lowerCAmelCase__ :    int				      =			0.02
											lowerCAmelCase__ :    Union[str, Any]				      =			0.03
											lowerCAmelCase__ :    str				      =			0.0_005
											lowerCAmelCase__ :    Dict				      =			0.000_001
											lowerCAmelCase__ :    Tuple				      =			1
											lowerCAmelCase__ :    Optional[int]				      =			10
											lowerCAmelCase__ :    int				      =			0.001
											lowerCAmelCase__ :    Union[str, Any]				      =			0.1
											lowerCAmelCase__ :    Tuple				      =			0.2
											lowerCAmelCase__ :    Optional[int]				      =			True

											# Model card mush be rendered and saved
											diffusers.utils.hub_utils.create_model_card(SCREAMING_SNAKE_CASE__ ,	model_name='''model_name'''    )
											self.assertTrue((Path(SCREAMING_SNAKE_CASE__    ) / '''README.md''').is_file()    )



