razvan commited on
Commit
250b7b4
·
verified ·
1 Parent(s): 37098fa

Upload plugins/mlintern/skills/ml-intern-harness/scripts/preflight_check.py with huggingface_hub

Browse files
plugins/mlintern/skills/ml-intern-harness/scripts/preflight_check.py ADDED
@@ -0,0 +1,35 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ #!/usr/bin/env python3
2
+ """Print an HF Jobs preflight checklist for ML Intern Codex tasks."""
3
+
4
+ from __future__ import annotations
5
+
6
+ import argparse
7
+
8
+
9
+ ITEMS = [
10
+ "Private GitHub repo exists or creation plan recorded",
11
+ "GitHub source branch/commit/PR identified",
12
+ "Reference implementation/docs used",
13
+ "Dataset repo/config/splits/columns verified",
14
+ "Model repo/tokenizer/license/gating verified",
15
+ "Smoke test completed",
16
+ "Hardware choice and timeout justified",
17
+ "Hub artifact output configured: push_to_hub or explicit upload",
18
+ "Monitoring configured: Trackio/dashboard/logged metrics",
19
+ "One-job-first plan for risky scripts or sweeps",
20
+ ]
21
+
22
+
23
+ def main() -> None:
24
+ parser = argparse.ArgumentParser(description=__doc__)
25
+ parser.add_argument("--task", default="<task>", help="Short task name")
26
+ parser.add_argument("--job", default="<job name>", help="Planned HF Job name")
27
+ args = parser.parse_args()
28
+
29
+ print(f"HF Jobs preflight for {args.task} ({args.job})")
30
+ for item in ITEMS:
31
+ print(f"- [ ] {item}:")
32
+
33
+
34
+ if __name__ == "__main__":
35
+ main()