mirror of
https://github.com/aljazceru/gpt-engineer.git
synced 2025-12-17 04:35:17 +01:00
f159dc45f2
* First step in collecting learnings * Rename prompts * remove requirements, use pip install -e . instead * Add requirements * Fix tests
33 lines
760 B
Python
33 lines
760 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 in ["prompt", "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)
|