githubmirror/gpt-engineer

scripts/clean_benchmarks.py

32 строки748 B

Patilla Code

make pre commit pass in the whole codebase (#149)

18 июн 2023, 15:46

18 июн 2023, 15:46084ce17

# 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)