Spaces:
Runtime error
Runtime error
File size: 1,833 Bytes
c87c295 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 |
if __name__ == "__main__":
import argparse
import os
import re
from zipfile import ZipFile
parser = argparse.ArgumentParser(
description="Package and zip LLM provided solutions"
)
parser.add_argument("--root", type=str, help="Root directory of solutions")
parser.add_argument(
"--output", type=str, required=True, help="Output directory of zip files"
)
args = parser.parse_args()
# [model_name]_temp_[temp] (without -sanitized)
directory_pattern = re.compile(r"(.*)_temp_(?:\d*\.?\d+)$")
assert os.path.isdir(args.root)
os.makedirs(args.output, exist_ok=True)
for directory in os.listdir(args.root):
match = directory_pattern.match(directory)
if not match:
continue
directory_name = match.group(0)
full_dir_path = os.path.join(args.root, directory_name)
assert os.path.isdir(full_dir_path)
print(f"Processing {full_dir_path}")
zip_file_path = os.path.join(args.output, f"{directory_name}.zip")
if os.path.exists(zip_file_path):
print(f"Skipping -- {zip_file_path} already exists")
continue
with ZipFile(zip_file_path, "w") as zip_file:
# directory_name/${TASK_ID}/${SAMPLE_ID}.py
for task_id in os.listdir(full_dir_path):
task_dir = os.path.join(full_dir_path, task_id)
if not os.path.isdir(task_dir):
continue
for sample_id in os.listdir(task_dir):
sample_file = os.path.join(task_dir, sample_id)
if not sample_file.endswith(".py"):
continue
zip_file.write(
sample_file, os.path.join(directory_name, task_id, sample_id)
)
|