mirror of
https://github.com/aljazceru/gpt-engineer.git
synced 2025-12-17 12:45:26 +01:00
33 lines
748 B
Python
33 lines
748 B
Python
# list all folders in benchmark folder
|
|
# for each folder, run the benchmark
|
|
|
|
import os
|
|
import shutil
|
|
|
|
from pathlib import Path
|
|
|
|
from typer import run
|
|
|
|
|
|
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)
|