mirror of https://github.com/coqui-ai/TTS.git
parent
38a29ce1c9
commit
4cc162a8b5
|
@ -1,20 +1,20 @@
|
|||
# #!/usr/bin/env python3`
|
||||
# import os
|
||||
# import shutil
|
||||
# import glob
|
||||
# from tests import get_tests_output_path
|
||||
# from TTS.utils.manage import ModelManager
|
||||
#!/usr/bin/env python3`
|
||||
import os
|
||||
import shutil
|
||||
import glob
|
||||
from tests import get_tests_output_path
|
||||
from TTS.utils.manage import ModelManager
|
||||
|
||||
# TODO: re-enable this after moving models to github releases
|
||||
# def test_if_all_models_available():
|
||||
# """Check if all the models are downloadable."""
|
||||
# print(" > Checking the availability of all the models under the ModelManager.")
|
||||
# manager = ModelManager(output_prefix=get_tests_output_path())
|
||||
# model_names = manager.list_models()
|
||||
# for model_name in model_names:
|
||||
# manager.download_model(model_name)
|
||||
# print(f" | > OK: {model_name}")
|
||||
|
||||
# folders = glob.glob(os.path.join(manager.output_prefix, '*'))
|
||||
# assert len(folders) == len(model_names)
|
||||
# shutil.rmtree(manager.output_prefix)
|
||||
def test_if_all_models_available():
|
||||
"""Check if all the models are downloadable."""
|
||||
print(" > Checking the availability of all the models under the ModelManager.")
|
||||
manager = ModelManager(output_prefix=get_tests_output_path())
|
||||
model_names = manager.list_models()
|
||||
for model_name in model_names:
|
||||
manager.download_model(model_name)
|
||||
print(f" | > OK: {model_name}")
|
||||
|
||||
folders = glob.glob(os.path.join(manager.output_prefix, '*'))
|
||||
assert len(folders) == len(model_names)
|
||||
shutil.rmtree(manager.output_prefix)
|
||||
|
|
Loading…
Reference in New Issue