02_generate_testset.py 515 Bytes
from __future__ import annotations

import sys

from _runner import print_artifacts, run_scripts


SCRIPTS = [
    "scripts/04_parse_docs.py",
    "scripts/05_generate_testset.py",
]

ARTIFACTS = [
    "data/parsed_docs/documents.jsonl",
    "data/parsed_docs/failed_parse.jsonl",
    "data/parsed_docs/parse_summary.json",
    "data/testsets/testset.raw.jsonl",
]


def main() -> int:
    code = run_scripts(SCRIPTS)
    print_artifacts(ARTIFACTS)
    return code


if __name__ == "__main__":
    sys.exit(main())