Commit 91743d9f authored by Leon Wu's avatar Leon Wu
Browse files

clear script

parent d5c60d39

clear_previous_results.py

+10 −5
Original line number Diff line number Diff line
@@ -7,8 +7,13 @@ results_folder = "../audiolm-pytorch-results" 
if not os.path.isdir(results_folder):

	raise AssertionError("didn't find results_folder, no results to clear out")



shutil.rmtree(f"{results_folder}/coarse_results")

shutil.rmtree(f"{results_folder}/fine_results")

shutil.rmtree(f"{results_folder}/semantic_results")

shutil.rmtree(f"{results_folder}/soundstream_results")

shutil.rmtree(f"{results_folder}/placeholder_dataset")
 
 No newline at end of file
 
for folder in [

	"coarse_results",

	"fine_results",

	"semantic_results",

	"soundstream_results",

	"placeholder_dataset",]:

	try:

	    shutil.rmtree(f"{results_folder}/{folder}")

	except FileNotFoundError:

	    pass  # directory does not exist, so do nothing