plugins/ml-intern/skills/ml-intern-harness/scripts/preflight_check.py

#!/usr/bin/env python3
"""Print an HF Jobs preflight checklist for ML Intern Codex tasks."""

from __future__ import annotations

import argparse


ITEMS = [
    "Private GitHub repo exists or creation plan recorded",
    "GitHub source branch/commit/PR identified",
    "Reference implementation/docs used",
    "Dataset repo/config/splits/columns verified",
    "Model repo/tokenizer/license/gating verified",
    "Smoke test completed",
    "Hardware choice and timeout justified",
    "Hub artifact output configured: push_to_hub or explicit upload",
    "Monitoring configured: Trackio/dashboard/logged metrics",
    "One-job-first plan for risky scripts or sweeps",
]


def main() -> None:
    parser = argparse.ArgumentParser(description=__doc__)
    parser.add_argument("--task", default="<task>", help="Short task name")
    parser.add_argument("--job", default="<job name>", help="Planned HF Job name")
    args = parser.parse_args()

    print(f"HF Jobs preflight for {args.task} ({args.job})")
    for item in ITEMS:
        print(f"- [ ] {item}:")


if __name__ == "__main__":
    main()