Medix-AI/coqui-tts
Medix-AI
/
coqui-tts
mirror of https://github.com/coqui-ai/TTS.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

disable model manager test until moving models to github releases

This commit is contained in:
Eren Gölge 2021-03-08 14:06:58 +01:00
parent 4e41dee178
commit dd4f46b60f
1 changed files with 18 additions and 18 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
tests/test_model_manager.py
View File

@ -1,20 +1,20 @@
#!/usr/bin/env python3` # #!/usr/bin/env python3`
import os # import os
import shutil # import shutil
import glob # import glob
from tests import get_tests_output_path # from tests import get_tests_output_path
from TTS.utils.manage import ModelManager # 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}")
def test_if_all_models_available(): # folders = glob.glob(os.path.join(manager.output_prefix, '*'))
"""Check if all the models are downloadable.""" # assert len(folders) == len(model_names)
print(" > Checking the availability of all the models under the ModelManager.") # shutil.rmtree(manager.output_prefix)
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)