Agent skill · NVIDIA
earth2studio-create-prognostic
Create Earth2Studio prognostic (time-stepping forecast) model wrappers. Do NOT use for diagnostic models, data sources, or installation.
What it needs
About 5k tokens when loaded.
What this skill does
Quick Start Checklist Do these steps IN ORDER. Do not skip any step. [ ] Read this SKILL.md completely first [ ] Get reference script (Step 0) [ ] Create earth2studio/models/px/<name>.py with triple inheritance [ ] Create test/models/px/test<name>.py with mock tests [ ] Run: uv run pytest test/models/px/test<name>.py -v [ ] Add/update model extra, install docs, API docs, and changelog (Steps 1-2, 9) [ ] Run: make format && make lint ⚠️ CRITICAL: Always use uv run for Python commands: - ✅ uv run pytest ... / uv run python ... - ❌ pytest ... / python ... (missing dependencies) Stuck or wrong output: Do not keep retrying the same fix. Follow Self-Improvement to patch this skill before continuing. Purpose Implement a prognostic model wrapper connecting third-party ML weather models to Earth2Studio. Prognostic models time-integrate forward—given initial state, they predict future states by stepping through time (e.g., 6-hour increments). Workspace Context Location --------- ---------- Harbor eval Write to /workspace/output/earth2studio/models/px/... Harbor + --copy-repo Full checkout at /workspace/repo Local clone Directory with pyproject.toml Never read evals/targets/ — grader references only. Reference Files Load on demand during the matching step: File Content Load at ------ --------- --------- references/skeleton-template.py Full model skeleton with FILL comments Steps 3–6 references/method-templates.py Canonical method implementations Steps 4–6 references/testing-guide.py Test skeleton and mock patterns Step 7 references/validation-guide.md Comparison scripts, PR, code review Steps 10–11 --- Workflow Steps Step 0 — Get Reference Script If $ARGUMENTS provided, use it. Otherwise ask: Please provide a reference inference script URL/path. Step 1 — Analyze & Propose Dependencies Analyze: packages, architecture, I/O shapes, time step, resolution, checkpoint. Propose pyproject.toml group (alphabetical, add to all). …
How to use it
Reference it in AdaL, Claude Code, Cursor or any coding agent — nothing to install:
@skills NVIDIA/earth2studio-create-prognostic