Function make_cache

evalplus/data/utils.py:27–45

(gzip_url, cache_path)

def make_cache(gzip_url, cache_path):
    # Check if human eval file exists in CACHE_DIR
    if not os.path.exists(cache_path):
        # Install HumanEval dataset and parse as jsonl
        print(f"Downloading dataset from {gzip_url}")
        with tempdir.TempDir() as tmpdir:
            plus_gz_path = os.path.join(tmpdir, f"data.jsonl.gz")
            wget.download(gzip_url, plus_gz_path)

            with gzip.open(plus_gz_path, "rb") as f:
                plus = f.read().decode("utf-8")

        # create CACHE_DIR if not exists
        if not os.path.exists(CACHE_DIR):
            os.makedirs(CACHE_DIR)

        # Write the original human eval file to CACHE_DIR
        with open(cache_path, "w") as f:
            f.write(plus)


def write_jsonl(