repro-bundle/v1/scripts/00_create_logbook.py

"""Create the Trackio logbook scaffold for the STE-quantization reproduction."""
import sys
from pathlib import Path

# IMPORTANT: run from the repro directory so .trackio lands inside it.
import trackio.logbook as lb

PROJ = Path(".trackio")

lb.create_logbook(title="Repro - High-Dimensional Learning Dynamics of Quantized Models with STE")

# Metadata: paper + tags (board picks these up).
lb.write_metadata(
    PROJ,
    {
        "paper": {
            "arxiv_id": "2510.10693",
            "openreview_id": "bI9moH3UZw",
            "title": "High-Dimensional Learning Dynamics of Quantized Models with Straight-Through Estimator",
            "authors": ["Yuma Ichikawa", "Shuhei Kashiwamura", "Ayaka Sakata"],
            "venue": "ICML 2026",
        },
        "tags": ["icml2026-repro", "paper-bI9moH3UZw"],
    },
)

# Pages (created empty; cells filled in by other scripts).
lb.add_page(PROJ, "Claim 1: ODE limit")
lb.add_page(PROJ, "Claim 2: Plateau + sharp drop")
lb.add_page(PROJ, "Claim 3: Fixed-point deviation")
lb.add_page(PROJ, "Methods")
lb.add_page(PROJ, "Conclusion")

# Scorecard page — first one, will pin the verdict cell to it.
lb.add_page(PROJ, "00-scorecard")

print("Logbook scaffolding created in:", PROJ.resolve())
print("Pages:", [p.name for p in (PROJ / "pages").iterdir() if p.is_dir()])