trainproof 0.3.0__tar.gz → 0.4.0__tar.gz

{trainproof-0.3.0/src/trainproof.egg-info → trainproof-0.4.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: trainproof
-Version: 0.3.0
+Version: 0.4.0
 Summary: A deterministic linter for ML training runs: dataset, tokenizer, and epoch logs.
 Author-email: "Panagiotis (Panos) Gkilis" <bedvibe@bedvibe.studio>
 License: MIT
@@ -79,12 +79,18 @@ famously fit random labels). **No single-run, loss-only rule can catch this
 class of failure** — its real signature is *relative*: a loss floor ~6x higher
 than a known-good run of the same task (5.59 vs 0.94).
 
-That finding sets the roadmap: v0.3 is `trainproof compare <run> <baseline>` —
-deterministic ratio rules against the healthy baseline you already have.
-See [ROADMAP.md](ROADMAP.md). (The gallery also improved v0.2 itself: the
-dead-run rule exists because `lr_zero` initially escaped with only a WARN.)
+That finding produced v0.3: `trainproof compare <run> <baseline>` —
+deterministic ratio rules against the healthy baseline you already have —
+which catches `bad_labels` at a 6x loss-floor ratio, in 3 seeds out of 3.
+The full study was repeated with three random seeds (15 runs):
+see [EVIDENCE_MATRIX.md](EVIDENCE_MATRIX.md) for every verdict, including the
+honest miss (compare alone overlooks one lr_zero seed — the single-run
+zero-LR fatality rule owns that case; the two commands cover each other's
+blind spots). The gallery also improved the tool itself twice: the dead-run
+rule and the total-zero-LR fatality rule both exist because runs escaped
+earlier rule versions. See [ROADMAP.md](ROADMAP.md).
 
-## The three commands
+## The commands
 
 ```bash
 # 1. Dataset preflight (speech/TTS pack): audio integrity, transcript quality,
@@ -116,9 +122,64 @@ TRAINPROOF VERDICT
 ========================================
 ```
 
+```
+
 Each command prints the verdict, writes a self-contained HTML report, and sets
 the process exit code — so it works as a CI gate out of the box.
 
+## Live guardian (v0.4)
+
+Don't wait for the post-mortem — catch a doomed run *while it is still burning
+GPU*. Add one line to a HuggingFace `Trainer`:
+
+```python
+from transformers import Trainer
+from trainproof.integrations.hf import TrainproofCallback
+
+trainer = Trainer(
+    ...,
+    callbacks=[TrainproofCallback(policy="stop_on_fail")],  # or policy="warn"
+)
+```
+
+Run against a real diverging QLoRA fine-tune (learning rate 100x too high), the
+guardian aborts it 20 steps into a 300-step schedule — on its own:
+
+```text
+{'loss': '1.784', 'grad_norm': '9.634',  'learning_rate': '0.007'}
+{'loss': '4.282', 'grad_norm': '53.76',  'learning_rate': '0.009'}
+{'loss': '10.6',  'grad_norm': '13.34',  'learning_rate': '0.011'}
+{'loss': '31.67', 'grad_norm': '76.67',  'learning_rate': '0.013'}
+...
+TRAINPROOF ABORT - stopping training at step 20. Findings:
+  [FAIL] Loss curve is diverging.
+         Evidence: End loss 22.952 vs Min loss 1.358
+  [FAIL] Loss never improved over the run (dead run).
+         Evidence: median of first 5 losses 1.502 vs last 5 22.952
+
+  scheduled steps : 300
+  stopped at step : 20
+  run saved       : 93% of the scheduled steps never ran
+```
+
+On a two-day pre-training run, that fraction is days of GPU time. Or watch a
+growing log file from outside the process (CI-friendly, exits non-zero on FAIL):
+
+```bash
+trainproof watch logs/run.jsonl --interval 10 --until-fail
+# [21:37:44] warming up (5 records)
+# [21:37:44] n_records=15 verdict=PASS findings=1
+```
+
+**The default is safe.** `policy="warn"` (the default) only observes and reports
+— it never interrupts your run, so you can leave it on even for experiments you
+expect to fail. Aborting is strictly opt-in via `policy="stop_on_fail"`, the one
+mode that takes an irreversible action. trainproof does not make that decision
+for you unless you ask.
+
+The guardian applies the same deterministic rules as `trainproof epoch`, so it
+inherits their documented single-run limitations.
+
 ## Supported log formats
 
 - HuggingFace Trainer (`trainer_state.json`)

trainproof-0.3.0/PKG-INFO → trainproof-0.4.0/README.md

@@ -1,153 +1,202 @@
-Metadata-Version: 2.4
-Name: trainproof
-Version: 0.3.0
-Summary: A deterministic linter for ML training runs: dataset, tokenizer, and epoch logs.
-Author-email: "Panagiotis (Panos) Gkilis" <bedvibe@bedvibe.studio>
-License: MIT
-Requires-Python: >=3.10
-Description-Content-Type: text/markdown
-Requires-Dist: numpy
-Requires-Dist: soundfile
-Requires-Dist: ttsproof
-
-# trainproof
-
-[![PyPI](https://img.shields.io/pypi/v/trainproof)](https://pypi.org/project/trainproof/)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
-
-**A deterministic linter for ML training runs.** Point it at your dataset, your
-tokenizer, or your first-epoch logs — it returns a PASS / WARN / FAIL verdict
-with named findings and cited evidence, before you burn days of GPU time on a
-run that was doomed at step 50.
-
-No ML judging ML. No invented "confidence 97%". Every rule is a deterministic
-threshold in [one auditable module](src/trainproof/rules.py), and every finding
-cites the numbers that triggered it.
-
-```bash
-pip install trainproof
-```
-
-## See a verdict in 60 seconds
-
-This repo ships the real logs of five QLoRA fine-tuning runs (Qwen2.5-3B,
-RTX 5080 — see the gallery below). Judge one right now:
-
-```bash
-trainproof epoch examples/gallery/lr_hot/trainer_state.json --format hf
-```
-
-```text
-[FAIL] Critical checks failed:
-  [FAIL] Loss curve is diverging.
-         Evidence: End loss 7.492 vs Min loss 1.398
-  [WARN] Gradient norm spikes detected.
-         Evidence: Max gn 2649.75 > 10.0x median (0.55)
-```
-
-## Why this exists
-
-The author lost a real 11-hour fine-tune to a failure nothing warned about.
-Pointed retroactively at that run's 1MB Coqui Trainer log (2,501 logged steps),
-trainproof's verdict: **FAIL — diverging**. The loss reached its minimum at 82%
-of the run and ended 1.9x above it. Translation: the final two hours of GPU
-time made the model measurably worse, and the checkpoint worth keeping had
-already existed for hours. No tool in the stack said a word.
-
-## The fault-injection gallery
-
-To validate the rules, the same QLoRA fine-tune (Qwen2.5-3B-Instruct, 4-bit,
-LoRA r=16, 300 steps on Alpaca-cleaned) was run five times — once healthy, four
-times with exactly one knob deliberately broken. Real runs, real logs, all
-shipped in [`examples/gallery/`](examples/gallery/):
-
-| Run | Sabotage | Verdict | Key evidence |
-|---|---|---|---|
-| `healthy` | none | **PASS** | loss 1.52 → 0.94, stable gradients |
-| `lr_hot` | LR x100 (2e-2) | **FAIL** | diverging: end 7.49 vs min 1.40; grad spike 2650 vs median 0.55 |
-| `lr_zero` | LR = 0 | **FAIL** | dead run: first-5 median 1.52 vs last-5 1.49 (<5% improvement); lr=0 on 100% of steps |
-| `fp16_nan` | fp16 + hot LR, no clipping | **FAIL** | diverging: end 7.21 vs min 1.09 (grad scaling absorbed the intended NaN — the run diverged instead; reported as observed) |
-| `bad_labels` | labels shuffled per-sequence | **WARN only** (single-run) — caught by `trainproof compare` (v0.3) | grad spike 23.3 vs median 1.09 |
-
-### The honest finding: loss curves cannot see corrupted data
-
-The `bad_labels` run — whose shuffled labels make real learning impossible —
-*reduced its loss by 62%* (18.9 → 5.75). The model was genuinely learning: not
-the task, but the marginal token statistics of the garbage. From its own loss
-curve, that is indistinguishable from healthy training (neural networks
-famously fit random labels). **No single-run, loss-only rule can catch this
-class of failure** — its real signature is *relative*: a loss floor ~6x higher
-than a known-good run of the same task (5.59 vs 0.94).
-
-That finding sets the roadmap: v0.3 is `trainproof compare <run> <baseline>` —
-deterministic ratio rules against the healthy baseline you already have.
-See [ROADMAP.md](ROADMAP.md). (The gallery also improved v0.2 itself: the
-dead-run rule exists because `lr_zero` initially escaped with only a WARN.)
-
-## The three commands
-
-```bash
-# 1. Dataset preflight (speech/TTS pack): audio integrity, transcript quality,
-#    duplicates, text-vs-audio duration mismatches
-trainproof data /path/to/dataset_or_manifest.jsonl
-
-# 2. Tokenizer preflight: vocabulary coverage, OOV rate, sequence blowouts,
-#    suspicious splits on numbers/dates
-trainproof tokenizer my_tokenizer.model transcripts.txt
-
-# 3. Training-run verdict: NaN/divergence/dead-run detection, gradient spikes,
-#    LR sanity, throughput — from log files, any framework
-trainproof epoch logs/run.jsonl            # exit code 1 on FAIL: CI-ready
-
-# 4. Compare against a baseline
-#    Catch relative pathologies like the `bad_labels` run that evade single-run rules.
-trainproof compare examples/gallery/bad_labels/trainer_state.json examples/gallery/healthy/trainer_state.json
-```
-
-```text
-========================================
-TRAINPROOF VERDICT
-========================================
-[FAIL] Critical checks failed:
-  [FAIL] loss floor ratio exceeded limit
-         Evidence: Run floor 5.592 vs Baseline floor 0.937 (ratio 6.0x > 2.0)
-  [FAIL] end loss ratio exceeded limit
-         Evidence: Run end 5.750 vs Baseline end 1.082 (ratio 5.3x > 2.0)
-========================================
-```
-
-Each command prints the verdict, writes a self-contained HTML report, and sets
-the process exit code — so it works as a CI gate out of the box.
-
-## Supported log formats
-
-- HuggingFace Trainer (`trainer_state.json`)
-- Coqui Trainer text logs (ANSI-colored `trainer_0_log.txt`)
-- Generic JSONL / CSV (columns: step, loss, lr, grad_norm, time — all optional)
-
-Auto-detected; override with `--format hf|coqui|jsonl|csv`. TensorBoard event
-files are planned (Lightning console captures are TTY dumps, not logs, and
-will not be supported).
-
-## Philosophy
-
-1. **Deterministic.** A rule fires or it doesn't. Thresholds live in one
-   module, commented, tunable.
-2. **Evidence-cited.** Every finding names the steps and values that triggered
-   it.
-3. **Honest about limits.** What the tool cannot detect is documented in the
-   README, not discovered by the user in production.
-
-## Family
-
-trainproof judges training runs. Its sibling [ttsproof](https://github.com/Mormolykos/ttsproof)
-judges TTS model *outputs* (structural audio checks, equivalence-aware WER/CER,
-published method with DOI) — and trainproof builds on it for the speech
-dataset checks.
-
-## Author
-
-Panagiotis (Panos) Gkilis — [portfolio](https://tts.bedvibe.studio/portfolio/) ·
-[bedvibe.studio](https://bedvibe.studio/)
-
-MIT license.
+# trainproof
+
+[![PyPI](https://img.shields.io/pypi/v/trainproof)](https://pypi.org/project/trainproof/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+
+**A deterministic linter for ML training runs.** Point it at your dataset, your
+tokenizer, or your first-epoch logs — it returns a PASS / WARN / FAIL verdict
+with named findings and cited evidence, before you burn days of GPU time on a
+run that was doomed at step 50.
+
+No ML judging ML. No invented "confidence 97%". Every rule is a deterministic
+threshold in [one auditable module](src/trainproof/rules.py), and every finding
+cites the numbers that triggered it.
+
+```bash
+pip install trainproof
+```
+
+## See a verdict in 60 seconds
+
+This repo ships the real logs of five QLoRA fine-tuning runs (Qwen2.5-3B,
+RTX 5080 — see the gallery below). Judge one right now:
+
+```bash
+trainproof epoch examples/gallery/lr_hot/trainer_state.json --format hf
+```
+
+```text
+[FAIL] Critical checks failed:
+  [FAIL] Loss curve is diverging.
+         Evidence: End loss 7.492 vs Min loss 1.398
+  [WARN] Gradient norm spikes detected.
+         Evidence: Max gn 2649.75 > 10.0x median (0.55)
+```
+
+## Why this exists
+
+The author lost a real 11-hour fine-tune to a failure nothing warned about.
+Pointed retroactively at that run's 1MB Coqui Trainer log (2,501 logged steps),
+trainproof's verdict: **FAIL — diverging**. The loss reached its minimum at 82%
+of the run and ended 1.9x above it. Translation: the final two hours of GPU
+time made the model measurably worse, and the checkpoint worth keeping had
+already existed for hours. No tool in the stack said a word.
+
+## The fault-injection gallery
+
+To validate the rules, the same QLoRA fine-tune (Qwen2.5-3B-Instruct, 4-bit,
+LoRA r=16, 300 steps on Alpaca-cleaned) was run five times — once healthy, four
+times with exactly one knob deliberately broken. Real runs, real logs, all
+shipped in [`examples/gallery/`](examples/gallery/):
+
+| Run | Sabotage | Verdict | Key evidence |
+|---|---|---|---|
+| `healthy` | none | **PASS** | loss 1.52 → 0.94, stable gradients |
+| `lr_hot` | LR x100 (2e-2) | **FAIL** | diverging: end 7.49 vs min 1.40; grad spike 2650 vs median 0.55 |
+| `lr_zero` | LR = 0 | **FAIL** | dead run: first-5 median 1.52 vs last-5 1.49 (<5% improvement); lr=0 on 100% of steps |
+| `fp16_nan` | fp16 + hot LR, no clipping | **FAIL** | diverging: end 7.21 vs min 1.09 (grad scaling absorbed the intended NaN — the run diverged instead; reported as observed) |
+| `bad_labels` | labels shuffled per-sequence | **WARN only** (single-run) — caught by `trainproof compare` (v0.3) | grad spike 23.3 vs median 1.09 |
+
+### The honest finding: loss curves cannot see corrupted data
+
+The `bad_labels` run — whose shuffled labels make real learning impossible —
+*reduced its loss by 62%* (18.9 → 5.75). The model was genuinely learning: not
+the task, but the marginal token statistics of the garbage. From its own loss
+curve, that is indistinguishable from healthy training (neural networks
+famously fit random labels). **No single-run, loss-only rule can catch this
+class of failure** — its real signature is *relative*: a loss floor ~6x higher
+than a known-good run of the same task (5.59 vs 0.94).
+
+That finding produced v0.3: `trainproof compare <run> <baseline>` —
+deterministic ratio rules against the healthy baseline you already have —
+which catches `bad_labels` at a 6x loss-floor ratio, in 3 seeds out of 3.
+The full study was repeated with three random seeds (15 runs):
+see [EVIDENCE_MATRIX.md](EVIDENCE_MATRIX.md) for every verdict, including the
+honest miss (compare alone overlooks one lr_zero seed — the single-run
+zero-LR fatality rule owns that case; the two commands cover each other's
+blind spots). The gallery also improved the tool itself twice: the dead-run
+rule and the total-zero-LR fatality rule both exist because runs escaped
+earlier rule versions. See [ROADMAP.md](ROADMAP.md).
+
+## The commands
+
+```bash
+# 1. Dataset preflight (speech/TTS pack): audio integrity, transcript quality,
+#    duplicates, text-vs-audio duration mismatches
+trainproof data /path/to/dataset_or_manifest.jsonl
+
+# 2. Tokenizer preflight: vocabulary coverage, OOV rate, sequence blowouts,
+#    suspicious splits on numbers/dates
+trainproof tokenizer my_tokenizer.model transcripts.txt
+
+# 3. Training-run verdict: NaN/divergence/dead-run detection, gradient spikes,
+#    LR sanity, throughput — from log files, any framework
+trainproof epoch logs/run.jsonl            # exit code 1 on FAIL: CI-ready
+
+# 4. Compare against a baseline
+#    Catch relative pathologies like the `bad_labels` run that evade single-run rules.
+trainproof compare examples/gallery/bad_labels/trainer_state.json examples/gallery/healthy/trainer_state.json
+```
+
+```text
+========================================
+TRAINPROOF VERDICT
+========================================
+[FAIL] Critical checks failed:
+  [FAIL] loss floor ratio exceeded limit
+         Evidence: Run floor 5.592 vs Baseline floor 0.937 (ratio 6.0x > 2.0)
+  [FAIL] end loss ratio exceeded limit
+         Evidence: Run end 5.750 vs Baseline end 1.082 (ratio 5.3x > 2.0)
+========================================
+```
+
+```
+
+Each command prints the verdict, writes a self-contained HTML report, and sets
+the process exit code — so it works as a CI gate out of the box.
+
+## Live guardian (v0.4)
+
+Don't wait for the post-mortem — catch a doomed run *while it is still burning
+GPU*. Add one line to a HuggingFace `Trainer`:
+
+```python
+from transformers import Trainer
+from trainproof.integrations.hf import TrainproofCallback
+
+trainer = Trainer(
+    ...,
+    callbacks=[TrainproofCallback(policy="stop_on_fail")],  # or policy="warn"
+)
+```
+
+Run against a real diverging QLoRA fine-tune (learning rate 100x too high), the
+guardian aborts it 20 steps into a 300-step schedule — on its own:
+
+```text
+{'loss': '1.784', 'grad_norm': '9.634',  'learning_rate': '0.007'}
+{'loss': '4.282', 'grad_norm': '53.76',  'learning_rate': '0.009'}
+{'loss': '10.6',  'grad_norm': '13.34',  'learning_rate': '0.011'}
+{'loss': '31.67', 'grad_norm': '76.67',  'learning_rate': '0.013'}
+...
+TRAINPROOF ABORT - stopping training at step 20. Findings:
+  [FAIL] Loss curve is diverging.
+         Evidence: End loss 22.952 vs Min loss 1.358
+  [FAIL] Loss never improved over the run (dead run).
+         Evidence: median of first 5 losses 1.502 vs last 5 22.952
+
+  scheduled steps : 300
+  stopped at step : 20
+  run saved       : 93% of the scheduled steps never ran
+```
+
+On a two-day pre-training run, that fraction is days of GPU time. Or watch a
+growing log file from outside the process (CI-friendly, exits non-zero on FAIL):
+
+```bash
+trainproof watch logs/run.jsonl --interval 10 --until-fail
+# [21:37:44] warming up (5 records)
+# [21:37:44] n_records=15 verdict=PASS findings=1
+```
+
+**The default is safe.** `policy="warn"` (the default) only observes and reports
+— it never interrupts your run, so you can leave it on even for experiments you
+expect to fail. Aborting is strictly opt-in via `policy="stop_on_fail"`, the one
+mode that takes an irreversible action. trainproof does not make that decision
+for you unless you ask.
+
+The guardian applies the same deterministic rules as `trainproof epoch`, so it
+inherits their documented single-run limitations.
+
+## Supported log formats
+
+- HuggingFace Trainer (`trainer_state.json`)
+- Coqui Trainer text logs (ANSI-colored `trainer_0_log.txt`)
+- Generic JSONL / CSV (columns: step, loss, lr, grad_norm, time — all optional)
+
+Auto-detected; override with `--format hf|coqui|jsonl|csv`. TensorBoard event
+files are planned (Lightning console captures are TTY dumps, not logs, and
+will not be supported).
+
+## Philosophy
+
+1. **Deterministic.** A rule fires or it doesn't. Thresholds live in one
+   module, commented, tunable.
+2. **Evidence-cited.** Every finding names the steps and values that triggered
+   it.
+3. **Honest about limits.** What the tool cannot detect is documented in the
+   README, not discovered by the user in production.
+
+## Family
+
+trainproof judges training runs. Its sibling [ttsproof](https://github.com/Mormolykos/ttsproof)
+judges TTS model *outputs* (structural audio checks, equivalence-aware WER/CER,
+published method with DOI) — and trainproof builds on it for the speech
+dataset checks.
+
+## Author
+
+Panagiotis (Panos) Gkilis — [portfolio](https://tts.bedvibe.studio/portfolio/) ·
+[bedvibe.studio](https://bedvibe.studio/)
+
+MIT license.

{trainproof-0.3.0 → trainproof-0.4.0}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "trainproof"
-version = "0.3.0"
+version = "0.4.0"
 description = "A deterministic linter for ML training runs: dataset, tokenizer, and epoch logs."
 authors = [{name = "Panagiotis (Panos) Gkilis", email = "bedvibe@bedvibe.studio"}]
 readme = "README.md"

{trainproof-0.3.0 → trainproof-0.4.0}/src/trainproof/cli.py

@@ -4,7 +4,9 @@ from pathlib import Path
 from .speech.data import check_data
 from .speech.tokenizer import check_tokenizer
 from .epoch import check_epoch
+from .epoch import check_epoch
 from .compare import check_compare
+from .watch import watch_loop
 from .report import print_verdict_console, write_html_report
 
 def main():
@@ -31,6 +33,13 @@ def main():
     compare_parser.add_argument("baseline", type=str, help="Path to baseline log file")
     compare_parser.add_argument("--format", choices=["auto", "hf", "coqui", "jsonl", "csv"], default="auto", help="Log format override")
 
+    # Watch Command
+    watch_parser = subparsers.add_parser("watch", help="Live guardian: poll a training log.")
+    watch_parser.add_argument("logfile", type=str, help="Path to run log file")
+    watch_parser.add_argument("--interval", type=int, default=10, help="Polling interval in seconds")
+    watch_parser.add_argument("--format", choices=["auto", "hf", "coqui", "jsonl", "csv"], default="auto", help="Log format override")
+    watch_parser.add_argument("--until-fail", action="store_true", help="Exit with code 1 as soon as verdict becomes FAIL")
+
     args = parser.parse_args()
     
     report_dict = {}
@@ -51,6 +60,9 @@ def main():
         except Exception as e:
             print(f"Error checking compare: {e}")
             sys.exit(1)
+    elif args.command == "watch":
+        watch_loop(args.logfile, interval=args.interval, fmt=args.format, until_fail=args.until_fail)
+        sys.exit(0)
 
     print_verdict_console(report_dict.get("verdict", "FAIL"), report_dict.get("findings", []))
     

{trainproof-0.3.0 → trainproof-0.4.0}/src/trainproof/epoch.py

@@ -4,13 +4,12 @@ from typing import Any
 from . import rules
 from .adapters import parse_log_with_format
 
-def check_epoch(log_path: str | Path, fmt: str = "auto") -> dict[str, Any]:
-    records = parse_log_with_format(log_path, fmt)
+def check_records(records: list[dict]) -> dict[str, Any]:
     findings = []
     verdict = "PASS"
     
     if not records:
-        return {"verdict": "FAIL", "findings": [{"level": "FAIL", "message": "No valid log records found.", "evidence": str(log_path)}]}
+        return {"verdict": "FAIL", "findings": [{"level": "FAIL", "message": "No valid log records found.", "evidence": ""}]}
 
     # Find relevant keys
     def get_val(row, aliases):
@@ -112,3 +111,9 @@ def check_epoch(log_path: str | Path, fmt: str = "auto") -> dict[str, Any]:
         findings.append({"level": "PASS", "message": "Loss curve shows healthy shape, grad norms are stable.", "evidence": f"{len(valid_losses)} steps analyzed."})
 
     return {"verdict": verdict, "findings": findings}
+
+def check_epoch(log_path: str | Path, fmt: str = "auto") -> dict[str, Any]:
+    records = parse_log_with_format(log_path, fmt)
+    if not records:
+        return {"verdict": "FAIL", "findings": [{"level": "FAIL", "message": "No valid log records found.", "evidence": str(log_path)}]}
+    return check_records(records)

trainproof-0.4.0/src/trainproof/integrations/__init__.py

@@ -0,0 +1,3 @@
+from .hf import TrainproofCallback
+
+__all__ = ["TrainproofCallback"]

trainproof-0.4.0/src/trainproof/integrations/hf.py

@@ -0,0 +1,89 @@
+from trainproof.epoch import check_records
+
+def _convert_state_to_records(state) -> list[dict]:
+    history = getattr(state, "log_history", [])
+    records = []
+    for entry in history:
+        if not isinstance(entry, dict) or "loss" not in entry:
+            continue
+        record = {}
+        for k in ["loss", "grad_norm", "step"]:
+            if k in entry and entry[k] is not None:
+                try:
+                    record[k] = float(entry[k])
+                except (ValueError, TypeError):
+                    pass
+        if "learning_rate" in entry and entry["learning_rate"] is not None:
+            try:
+                record["lr"] = float(entry["learning_rate"])
+            except (ValueError, TypeError):
+                pass
+        records.append(record)
+    return records
+
+try:
+    from transformers import TrainerCallback
+except ImportError:
+    class TrainerCallback:
+        pass
+    _HAS_TRANSFORMERS = False
+else:
+    _HAS_TRANSFORMERS = True
+
+
+class TrainproofCallback(TrainerCallback):
+    """Judge a live HuggingFace training run with trainproof's deterministic rules.
+
+    policy:
+      "warn" (DEFAULT): observe and report only. On a FAIL verdict it prints the
+        findings and lets training continue. It NEVER interrupts your run — safe
+        to leave on by default, including for experiments you expect to fail.
+      "stop_on_fail": opt-in. Additionally sets control.should_training_stop on a
+        FAIL verdict, aborting the run to save GPU time. This is the only mode
+        that takes an irreversible action, and you must ask for it explicitly.
+
+    check_every: minimum steps between checks. min_points: minimum logged loss
+    points before any verdict is issued (avoids judging warm-up noise).
+    A FAIL is announced once, not re-announced every subsequent check.
+    """
+
+    def __init__(self, policy="warn", check_every=25, min_points=10):
+        if not _HAS_TRANSFORMERS:
+            raise ImportError("pip install transformers is required to use TrainproofCallback")
+        self.policy = policy
+        self.check_every = check_every
+        self.min_points = min_points
+        self.last_checked_step = 0
+        self.last_verdict = None
+
+    def on_log(self, args, state, control, **kwargs):
+        step = getattr(state, "global_step", 0)
+        if step < self.last_checked_step + self.check_every:
+            return
+
+        records = _convert_state_to_records(state)
+        if len(records) < self.min_points:
+            return
+
+        self.last_checked_step = step
+        
+        report = check_records(records)
+        verdict = report.get("verdict", "UNKNOWN")
+        findings = report.get("findings", [])
+
+        if verdict == "FAIL" and self.last_verdict != "FAIL":
+            if self.policy == "stop_on_fail":
+                setattr(control, "should_training_stop", True)
+                print(f"\nTRAINPROOF ABORT - stopping training at step {step}. Findings:")
+            else:
+                print("\nTRAINPROOF WARNING - this run looks doomed:")
+                
+            for f in findings:
+                level = f.get("level", "INFO")
+                msg = f.get("message", "")
+                ev = f.get("evidence", "")
+                print(f"  [{level}] {msg}")
+                if ev:
+                    print(f"         Evidence: {ev}")
+
+        self.last_verdict = verdict

trainproof-0.4.0/src/trainproof/watch.py

@@ -0,0 +1,57 @@
+import time
+import sys
+from datetime import datetime
+from pathlib import Path
+from .adapters import parse_log_with_format
+from .epoch import check_records
+from .report import print_verdict_console
+
+def poll_once(path: str | Path, fmt: str, prev_verdict: str | None) -> tuple[str | None, bool, int]:
+    try:
+        records = parse_log_with_format(path, fmt)
+    except Exception:
+        records = []
+        
+    n_records = len(records)
+    
+    valid_losses = 0
+    for r in records:
+        if "loss" in r and r["loss"] is not None:
+            valid_losses += 1
+            
+    now_str = datetime.now().strftime("%H:%M:%S")
+    
+    if valid_losses < 10:
+        print(f"[{now_str}] warming up ({n_records} records)")
+        return (None, False, n_records)
+        
+    report = check_records(records)
+    verdict = report.get("verdict", "UNKNOWN")
+    n_findings = len(report.get("findings", []))
+    
+    changed = (verdict != prev_verdict)
+    
+    print(f"[{now_str}] n_records={n_records} verdict={verdict} findings={n_findings}")
+    
+    if changed:
+        print_verdict_console(verdict, report.get("findings", []))
+        
+    return (verdict, changed, n_records)
+
+def watch_loop(path: str | Path, interval: int = 10, fmt: str = "auto", until_fail: bool = False):
+    prev_verdict = None
+    try:
+        while True:
+            verdict, changed, n = poll_once(path, fmt, prev_verdict)
+            if verdict is not None:
+                prev_verdict = verdict
+            
+            if until_fail and verdict == "FAIL":
+                sys.exit(1)
+                
+            time.sleep(interval)
+    except KeyboardInterrupt:
+        print(f"\nStopped watching. Final verdict: {prev_verdict}")
+        if prev_verdict == "FAIL":
+            sys.exit(1)
+        sys.exit(0)

trainproof-0.3.0/README.md → trainproof-0.4.0/src/trainproof.egg-info/PKG-INFO

@@ -1,141 +1,214 @@
-# trainproof
-
-[![PyPI](https://img.shields.io/pypi/v/trainproof)](https://pypi.org/project/trainproof/)
-[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
-
-**A deterministic linter for ML training runs.** Point it at your dataset, your
-tokenizer, or your first-epoch logs — it returns a PASS / WARN / FAIL verdict
-with named findings and cited evidence, before you burn days of GPU time on a
-run that was doomed at step 50.
-
-No ML judging ML. No invented "confidence 97%". Every rule is a deterministic
-threshold in [one auditable module](src/trainproof/rules.py), and every finding
-cites the numbers that triggered it.
-
-```bash
-pip install trainproof
-```
-
-## See a verdict in 60 seconds
-
-This repo ships the real logs of five QLoRA fine-tuning runs (Qwen2.5-3B,
-RTX 5080 — see the gallery below). Judge one right now:
-
-```bash
-trainproof epoch examples/gallery/lr_hot/trainer_state.json --format hf
-```
-
-```text
-[FAIL] Critical checks failed:
-  [FAIL] Loss curve is diverging.
-         Evidence: End loss 7.492 vs Min loss 1.398
-  [WARN] Gradient norm spikes detected.
-         Evidence: Max gn 2649.75 > 10.0x median (0.55)
-```
-
-## Why this exists
-
-The author lost a real 11-hour fine-tune to a failure nothing warned about.
-Pointed retroactively at that run's 1MB Coqui Trainer log (2,501 logged steps),
-trainproof's verdict: **FAIL — diverging**. The loss reached its minimum at 82%
-of the run and ended 1.9x above it. Translation: the final two hours of GPU
-time made the model measurably worse, and the checkpoint worth keeping had
-already existed for hours. No tool in the stack said a word.
-
-## The fault-injection gallery
-
-To validate the rules, the same QLoRA fine-tune (Qwen2.5-3B-Instruct, 4-bit,
-LoRA r=16, 300 steps on Alpaca-cleaned) was run five times — once healthy, four
-times with exactly one knob deliberately broken. Real runs, real logs, all
-shipped in [`examples/gallery/`](examples/gallery/):
-
-| Run | Sabotage | Verdict | Key evidence |
-|---|---|---|---|
-| `healthy` | none | **PASS** | loss 1.52 → 0.94, stable gradients |
-| `lr_hot` | LR x100 (2e-2) | **FAIL** | diverging: end 7.49 vs min 1.40; grad spike 2650 vs median 0.55 |
-| `lr_zero` | LR = 0 | **FAIL** | dead run: first-5 median 1.52 vs last-5 1.49 (<5% improvement); lr=0 on 100% of steps |
-| `fp16_nan` | fp16 + hot LR, no clipping | **FAIL** | diverging: end 7.21 vs min 1.09 (grad scaling absorbed the intended NaN — the run diverged instead; reported as observed) |
-| `bad_labels` | labels shuffled per-sequence | **WARN only** (single-run) — caught by `trainproof compare` (v0.3) | grad spike 23.3 vs median 1.09 |
-
-### The honest finding: loss curves cannot see corrupted data
-
-The `bad_labels` run — whose shuffled labels make real learning impossible —
-*reduced its loss by 62%* (18.9 → 5.75). The model was genuinely learning: not
-the task, but the marginal token statistics of the garbage. From its own loss
-curve, that is indistinguishable from healthy training (neural networks
-famously fit random labels). **No single-run, loss-only rule can catch this
-class of failure** — its real signature is *relative*: a loss floor ~6x higher
-than a known-good run of the same task (5.59 vs 0.94).
-
-That finding sets the roadmap: v0.3 is `trainproof compare <run> <baseline>` —
-deterministic ratio rules against the healthy baseline you already have.
-See [ROADMAP.md](ROADMAP.md). (The gallery also improved v0.2 itself: the
-dead-run rule exists because `lr_zero` initially escaped with only a WARN.)
-
-## The three commands
-
-```bash
-# 1. Dataset preflight (speech/TTS pack): audio integrity, transcript quality,
-#    duplicates, text-vs-audio duration mismatches
-trainproof data /path/to/dataset_or_manifest.jsonl
-
-# 2. Tokenizer preflight: vocabulary coverage, OOV rate, sequence blowouts,
-#    suspicious splits on numbers/dates
-trainproof tokenizer my_tokenizer.model transcripts.txt
-
-# 3. Training-run verdict: NaN/divergence/dead-run detection, gradient spikes,
-#    LR sanity, throughput — from log files, any framework
-trainproof epoch logs/run.jsonl            # exit code 1 on FAIL: CI-ready
-
-# 4. Compare against a baseline
-#    Catch relative pathologies like the `bad_labels` run that evade single-run rules.
-trainproof compare examples/gallery/bad_labels/trainer_state.json examples/gallery/healthy/trainer_state.json
-```
-
-```text
-========================================
-TRAINPROOF VERDICT
-========================================
-[FAIL] Critical checks failed:
-  [FAIL] loss floor ratio exceeded limit
-         Evidence: Run floor 5.592 vs Baseline floor 0.937 (ratio 6.0x > 2.0)
-  [FAIL] end loss ratio exceeded limit
-         Evidence: Run end 5.750 vs Baseline end 1.082 (ratio 5.3x > 2.0)
-========================================
-```
-
-Each command prints the verdict, writes a self-contained HTML report, and sets
-the process exit code — so it works as a CI gate out of the box.
-
-## Supported log formats
-
-- HuggingFace Trainer (`trainer_state.json`)
-- Coqui Trainer text logs (ANSI-colored `trainer_0_log.txt`)
-- Generic JSONL / CSV (columns: step, loss, lr, grad_norm, time — all optional)
-
-Auto-detected; override with `--format hf|coqui|jsonl|csv`. TensorBoard event
-files are planned (Lightning console captures are TTY dumps, not logs, and
-will not be supported).
-
-## Philosophy
-
-1. **Deterministic.** A rule fires or it doesn't. Thresholds live in one
-   module, commented, tunable.
-2. **Evidence-cited.** Every finding names the steps and values that triggered
-   it.
-3. **Honest about limits.** What the tool cannot detect is documented in the
-   README, not discovered by the user in production.
-
-## Family
-
-trainproof judges training runs. Its sibling [ttsproof](https://github.com/Mormolykos/ttsproof)
-judges TTS model *outputs* (structural audio checks, equivalence-aware WER/CER,
-published method with DOI) — and trainproof builds on it for the speech
-dataset checks.
-
-## Author
-
-Panagiotis (Panos) Gkilis — [portfolio](https://tts.bedvibe.studio/portfolio/) ·
-[bedvibe.studio](https://bedvibe.studio/)
-
-MIT license.
+Metadata-Version: 2.4
+Name: trainproof
+Version: 0.4.0
+Summary: A deterministic linter for ML training runs: dataset, tokenizer, and epoch logs.
+Author-email: "Panagiotis (Panos) Gkilis" <bedvibe@bedvibe.studio>
+License: MIT
+Requires-Python: >=3.10
+Description-Content-Type: text/markdown
+Requires-Dist: numpy
+Requires-Dist: soundfile
+Requires-Dist: ttsproof
+
+# trainproof
+
+[![PyPI](https://img.shields.io/pypi/v/trainproof)](https://pypi.org/project/trainproof/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](LICENSE)
+
+**A deterministic linter for ML training runs.** Point it at your dataset, your
+tokenizer, or your first-epoch logs — it returns a PASS / WARN / FAIL verdict
+with named findings and cited evidence, before you burn days of GPU time on a
+run that was doomed at step 50.
+
+No ML judging ML. No invented "confidence 97%". Every rule is a deterministic
+threshold in [one auditable module](src/trainproof/rules.py), and every finding
+cites the numbers that triggered it.
+
+```bash
+pip install trainproof
+```
+
+## See a verdict in 60 seconds
+
+This repo ships the real logs of five QLoRA fine-tuning runs (Qwen2.5-3B,
+RTX 5080 — see the gallery below). Judge one right now:
+
+```bash
+trainproof epoch examples/gallery/lr_hot/trainer_state.json --format hf
+```
+
+```text
+[FAIL] Critical checks failed:
+  [FAIL] Loss curve is diverging.
+         Evidence: End loss 7.492 vs Min loss 1.398
+  [WARN] Gradient norm spikes detected.
+         Evidence: Max gn 2649.75 > 10.0x median (0.55)
+```
+
+## Why this exists
+
+The author lost a real 11-hour fine-tune to a failure nothing warned about.
+Pointed retroactively at that run's 1MB Coqui Trainer log (2,501 logged steps),
+trainproof's verdict: **FAIL — diverging**. The loss reached its minimum at 82%
+of the run and ended 1.9x above it. Translation: the final two hours of GPU
+time made the model measurably worse, and the checkpoint worth keeping had
+already existed for hours. No tool in the stack said a word.
+
+## The fault-injection gallery
+
+To validate the rules, the same QLoRA fine-tune (Qwen2.5-3B-Instruct, 4-bit,
+LoRA r=16, 300 steps on Alpaca-cleaned) was run five times — once healthy, four
+times with exactly one knob deliberately broken. Real runs, real logs, all
+shipped in [`examples/gallery/`](examples/gallery/):
+
+| Run | Sabotage | Verdict | Key evidence |
+|---|---|---|---|
+| `healthy` | none | **PASS** | loss 1.52 → 0.94, stable gradients |
+| `lr_hot` | LR x100 (2e-2) | **FAIL** | diverging: end 7.49 vs min 1.40; grad spike 2650 vs median 0.55 |
+| `lr_zero` | LR = 0 | **FAIL** | dead run: first-5 median 1.52 vs last-5 1.49 (<5% improvement); lr=0 on 100% of steps |
+| `fp16_nan` | fp16 + hot LR, no clipping | **FAIL** | diverging: end 7.21 vs min 1.09 (grad scaling absorbed the intended NaN — the run diverged instead; reported as observed) |
+| `bad_labels` | labels shuffled per-sequence | **WARN only** (single-run) — caught by `trainproof compare` (v0.3) | grad spike 23.3 vs median 1.09 |
+
+### The honest finding: loss curves cannot see corrupted data
+
+The `bad_labels` run — whose shuffled labels make real learning impossible —
+*reduced its loss by 62%* (18.9 → 5.75). The model was genuinely learning: not
+the task, but the marginal token statistics of the garbage. From its own loss
+curve, that is indistinguishable from healthy training (neural networks
+famously fit random labels). **No single-run, loss-only rule can catch this
+class of failure** — its real signature is *relative*: a loss floor ~6x higher
+than a known-good run of the same task (5.59 vs 0.94).
+
+That finding produced v0.3: `trainproof compare <run> <baseline>` —
+deterministic ratio rules against the healthy baseline you already have —
+which catches `bad_labels` at a 6x loss-floor ratio, in 3 seeds out of 3.
+The full study was repeated with three random seeds (15 runs):
+see [EVIDENCE_MATRIX.md](EVIDENCE_MATRIX.md) for every verdict, including the
+honest miss (compare alone overlooks one lr_zero seed — the single-run
+zero-LR fatality rule owns that case; the two commands cover each other's
+blind spots). The gallery also improved the tool itself twice: the dead-run
+rule and the total-zero-LR fatality rule both exist because runs escaped
+earlier rule versions. See [ROADMAP.md](ROADMAP.md).
+
+## The commands
+
+```bash
+# 1. Dataset preflight (speech/TTS pack): audio integrity, transcript quality,
+#    duplicates, text-vs-audio duration mismatches
+trainproof data /path/to/dataset_or_manifest.jsonl
+
+# 2. Tokenizer preflight: vocabulary coverage, OOV rate, sequence blowouts,
+#    suspicious splits on numbers/dates
+trainproof tokenizer my_tokenizer.model transcripts.txt
+
+# 3. Training-run verdict: NaN/divergence/dead-run detection, gradient spikes,
+#    LR sanity, throughput — from log files, any framework
+trainproof epoch logs/run.jsonl            # exit code 1 on FAIL: CI-ready
+
+# 4. Compare against a baseline
+#    Catch relative pathologies like the `bad_labels` run that evade single-run rules.
+trainproof compare examples/gallery/bad_labels/trainer_state.json examples/gallery/healthy/trainer_state.json
+```
+
+```text
+========================================
+TRAINPROOF VERDICT
+========================================
+[FAIL] Critical checks failed:
+  [FAIL] loss floor ratio exceeded limit
+         Evidence: Run floor 5.592 vs Baseline floor 0.937 (ratio 6.0x > 2.0)
+  [FAIL] end loss ratio exceeded limit
+         Evidence: Run end 5.750 vs Baseline end 1.082 (ratio 5.3x > 2.0)
+========================================
+```
+
+```
+
+Each command prints the verdict, writes a self-contained HTML report, and sets
+the process exit code — so it works as a CI gate out of the box.
+
+## Live guardian (v0.4)
+
+Don't wait for the post-mortem — catch a doomed run *while it is still burning
+GPU*. Add one line to a HuggingFace `Trainer`:
+
+```python
+from transformers import Trainer
+from trainproof.integrations.hf import TrainproofCallback
+
+trainer = Trainer(
+    ...,
+    callbacks=[TrainproofCallback(policy="stop_on_fail")],  # or policy="warn"
+)
+```
+
+Run against a real diverging QLoRA fine-tune (learning rate 100x too high), the
+guardian aborts it 20 steps into a 300-step schedule — on its own:
+
+```text
+{'loss': '1.784', 'grad_norm': '9.634',  'learning_rate': '0.007'}
+{'loss': '4.282', 'grad_norm': '53.76',  'learning_rate': '0.009'}
+{'loss': '10.6',  'grad_norm': '13.34',  'learning_rate': '0.011'}
+{'loss': '31.67', 'grad_norm': '76.67',  'learning_rate': '0.013'}
+...
+TRAINPROOF ABORT - stopping training at step 20. Findings:
+  [FAIL] Loss curve is diverging.
+         Evidence: End loss 22.952 vs Min loss 1.358
+  [FAIL] Loss never improved over the run (dead run).
+         Evidence: median of first 5 losses 1.502 vs last 5 22.952
+
+  scheduled steps : 300
+  stopped at step : 20
+  run saved       : 93% of the scheduled steps never ran
+```
+
+On a two-day pre-training run, that fraction is days of GPU time. Or watch a
+growing log file from outside the process (CI-friendly, exits non-zero on FAIL):
+
+```bash
+trainproof watch logs/run.jsonl --interval 10 --until-fail
+# [21:37:44] warming up (5 records)
+# [21:37:44] n_records=15 verdict=PASS findings=1
+```
+
+**The default is safe.** `policy="warn"` (the default) only observes and reports
+— it never interrupts your run, so you can leave it on even for experiments you
+expect to fail. Aborting is strictly opt-in via `policy="stop_on_fail"`, the one
+mode that takes an irreversible action. trainproof does not make that decision
+for you unless you ask.
+
+The guardian applies the same deterministic rules as `trainproof epoch`, so it
+inherits their documented single-run limitations.
+
+## Supported log formats
+
+- HuggingFace Trainer (`trainer_state.json`)
+- Coqui Trainer text logs (ANSI-colored `trainer_0_log.txt`)
+- Generic JSONL / CSV (columns: step, loss, lr, grad_norm, time — all optional)
+
+Auto-detected; override with `--format hf|coqui|jsonl|csv`. TensorBoard event
+files are planned (Lightning console captures are TTY dumps, not logs, and
+will not be supported).
+
+## Philosophy
+
+1. **Deterministic.** A rule fires or it doesn't. Thresholds live in one
+   module, commented, tunable.
+2. **Evidence-cited.** Every finding names the steps and values that triggered
+   it.
+3. **Honest about limits.** What the tool cannot detect is documented in the
+   README, not discovered by the user in production.
+
+## Family
+
+trainproof judges training runs. Its sibling [ttsproof](https://github.com/Mormolykos/ttsproof)
+judges TTS model *outputs* (structural audio checks, equivalence-aware WER/CER,
+published method with DOI) — and trainproof builds on it for the speech
+dataset checks.
+
+## Author
+
+Panagiotis (Panos) Gkilis — [portfolio](https://tts.bedvibe.studio/portfolio/) ·
+[bedvibe.studio](https://bedvibe.studio/)
+
+MIT license.

{trainproof-0.3.0 → trainproof-0.4.0}/src/trainproof.egg-info/SOURCES.txt

@@ -7,16 +7,21 @@ src/trainproof/compare.py
 src/trainproof/epoch.py
 src/trainproof/report.py
 src/trainproof/rules.py
+src/trainproof/watch.py
 src/trainproof.egg-info/PKG-INFO
 src/trainproof.egg-info/SOURCES.txt
 src/trainproof.egg-info/dependency_links.txt
 src/trainproof.egg-info/entry_points.txt
 src/trainproof.egg-info/requires.txt
 src/trainproof.egg-info/top_level.txt
+src/trainproof/integrations/__init__.py
+src/trainproof/integrations/hf.py
 src/trainproof/speech/__init__.py
 src/trainproof/speech/data.py
 src/trainproof/speech/tokenizer.py
 tests/test_adapters.py
 tests/test_compare.py
 tests/test_data.py
-tests/test_epoch.py
+tests/test_epoch.py
+tests/test_hf_callback.py
+tests/test_watch.py

trainproof-0.4.0/tests/test_hf_callback.py

@@ -0,0 +1,66 @@
+import trainproof.integrations.hf as hf_mod
+from trainproof.integrations.hf import _convert_state_to_records, TrainproofCallback
+
+# Patch so tests can run without transformers installed
+hf_mod._HAS_TRANSFORMERS = True
+class MockState:
+    def __init__(self):
+        self.global_step = 0
+        self.log_history = []
+
+class MockControl:
+    def __init__(self):
+        self.should_training_stop = False
+
+def test_convert_state_to_records():
+    state = MockState()
+    state.log_history = [
+        {"loss": 1.0, "step": 10, "learning_rate": 0.01, "eval_loss": 2.0},
+        {"eval_loss": 1.5, "step": 10}, 
+        {"loss": 0.5, "step": 20, "grad_norm": 0.1}
+    ]
+    records = _convert_state_to_records(state)
+    assert len(records) == 2
+    assert records[0] == {"loss": 1.0, "step": 10.0, "lr": 0.01}
+    assert records[1] == {"loss": 0.5, "step": 20.0, "grad_norm": 0.1}
+
+def test_hf_callback_warn_policy():
+    callback = TrainproofCallback(policy="warn", check_every=10, min_points=10)
+    state = MockState()
+    control = MockControl()
+    
+    # Add < 10 points -> should do nothing
+    for i in range(5):
+        state.log_history.append({"loss": 10.0, "step": i})
+    state.global_step = 10
+    callback.on_log(None, state, control)
+    assert callback.last_verdict is None
+    
+    # Add up to 15 points (healthy) -> PASS
+    for i in range(5, 15):
+        state.log_history.append({"loss": 1.0, "step": i, "learning_rate": 1e-4})
+    state.global_step = 20
+    callback.on_log(None, state, control)
+    assert callback.last_verdict == "PASS"
+    assert not control.should_training_stop
+    
+    # Diverge -> FAIL but don't stop because policy=warn
+    for i in range(15, 25):
+        state.log_history.append({"loss": 50.0, "step": i, "learning_rate": 1e-4})
+    state.global_step = 30
+    callback.on_log(None, state, control)
+    assert callback.last_verdict == "FAIL"
+    assert not control.should_training_stop
+
+def test_hf_callback_stop_on_fail():
+    callback = TrainproofCallback(policy="stop_on_fail", check_every=10, min_points=10)
+    state = MockState()
+    control = MockControl()
+    
+    for i in range(20):
+        state.log_history.append({"loss": 50.0, "step": i, "learning_rate": 1e-4}) 
+    state.global_step = 20
+    
+    callback.on_log(None, state, control)
+    assert callback.last_verdict == "FAIL"
+    assert control.should_training_stop

trainproof-0.4.0/tests/test_watch.py

@@ -0,0 +1,42 @@
+import json
+from pathlib import Path
+from trainproof.watch import poll_once
+
+def test_watch_poll_once(tmp_path: Path):
+    log_file = tmp_path / "test.jsonl"
+    
+    # Phase 1: warming up (less than 10 valid losses)
+    logs = []
+    for i in range(5):
+        logs.append(json.dumps({"loss": 5.0, "step": i, "lr": 1e-4}))
+    log_file.write_text("\n".join(logs))
+    
+    verdict, changed, n = poll_once(log_file, "jsonl", None)
+    assert verdict is None
+    assert not changed
+    assert n == 5
+    
+    # Phase 2: healthy phase -> PASS
+    for i in range(5, 15):
+        logs.append(json.dumps({"loss": 1.0, "step": i, "lr": 1e-4}))
+    log_file.write_text("\n".join(logs))
+    
+    verdict, changed, n = poll_once(log_file, "jsonl", None)
+    assert verdict == "PASS"
+    assert changed
+    assert n == 15
+    
+    # Check that it doesn't say "changed" if prev_verdict was PASS
+    verdict2, changed2, n2 = poll_once(log_file, "jsonl", "PASS")
+    assert verdict2 == "PASS"
+    assert not changed2
+    
+    # Phase 3: append diverging losses -> verdict changes to FAIL
+    for i in range(15, 25):
+        logs.append(json.dumps({"loss": 20.0, "step": i, "lr": 1e-4}))
+    log_file.write_text("\n".join(logs))
+    
+    verdict3, changed3, n3 = poll_once(log_file, "jsonl", "PASS")
+    assert verdict3 == "FAIL"
+    assert changed3
+    assert n3 == 25