diff --git a/scripts/clean_benchmarks.py b/scripts/clean_benchmarks.py new file mode 100644 index 0000000..21cb536 --- /dev/null +++ b/scripts/clean_benchmarks.py @@ -0,0 +1,38 @@ +# list all folders in benchmark folder +# for each folder, run the benchmark + +import os +import sys +import subprocess +import time +import datetime +import shutil +import argparse +import json +from pathlib import Path +from typer import run +from itertools import islice + +def main( +): + benchmarks = Path('benchmark') + + for benchmark in benchmarks.iterdir(): + if benchmark.is_dir(): + print(f'Cleaning {benchmark}') + for path in benchmark.iterdir(): + if path.name == 'main_prompt': + continue + + # Get filename of Path object + if path.is_dir(): + # delete the entire directory + shutil.rmtree(path) + else: + # delete the file + os.remove(path) + +if __name__ == '__main__': + run(main) + +