From c6df8de80aca83a860e590ef96c9eb6830b6c820 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Eren=20G=C3=B6lge?= Date: Thu, 6 May 2021 03:40:26 +0200 Subject: [PATCH] remove output train folder at the end of the test --- tests/tts_tests/test_speedy_speech_train.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/tests/tts_tests/test_speedy_speech_train.py b/tests/tts_tests/test_speedy_speech_train.py index 341174f7..1b356985 100644 --- a/tests/tts_tests/test_speedy_speech_train.py +++ b/tests/tts_tests/test_speedy_speech_train.py @@ -1,8 +1,8 @@ import glob import os +import shutil from tests import get_tests_output_path, run_cli -from TTS.config import BaseDatasetConfig from TTS.tts.configs import SpeedySpeechConfig config_path = os.path.join(get_tests_output_path(), "test_speedy_speech_config.json") @@ -46,3 +46,4 @@ continue_path = max(glob.glob(os.path.join(output_path, "*/")), key=os.path.getm # restore the model and continue training for one more epoch command_train = f"CUDA_VISIBLE_DEVICES='' python TTS/bin/train_speedy_speech.py --continue_path {continue_path} " run_cli(command_train) +shutil.rmtree(continue_path)