githubmirror/gpt-engineer

scripts/clean_benchmarks.py

32 строки760 B

Anton Osika

First step in making gpt-engineer learn. Rename main_prompt -> prompt (#381)

24 июн 2023, 18:53

24 июн 2023, 18:53f159dc4

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