Automate failure-driven training continuation

backend/tests/test_belgium_training_loop.py

@@ -62,6 +62,20 @@ def test_training_command_supports_conservative_aerial_finetuning(tmp_path: Path
     assert f"data={tmp_path / 'dataset.yaml'}" in command
 
 
+def test_failed_iteration_builds_train_only_sampling_for_next_checkpoint(tmp_path: Path) -> None:
+    command = MODULE.failure_sampling_command(
+        scripts_dir=tmp_path / "scripts",
+        train_summary=tmp_path / "train-summary.json",
+        corpus_manifest=tmp_path / "manifest.json",
+        assessment=tmp_path / "assessment.json",
+        output_dir=tmp_path / "iteration-001" / "failure-driven-training",
+    )
+    assert command[1].endswith("build_failure_driven_yolo_sampling.py")
+    assert command[command.index("--summary") + 1].endswith("train-summary.json")
+    assert command[command.index("--assessment") + 1].endswith("assessment.json")
+    assert command[command.index("--output-dir") + 1].endswith("failure-driven-training")
+
+
 def test_loop_refuses_failed_dataset_audit(tmp_path: Path) -> None:
     audit = tmp_path / "audit.json"
     audit.write_text(json.dumps({"status": "needs_attention", "low_variance_positive_tile_count": 4}))
@@ -73,6 +87,8 @@ def test_loop_refuses_failed_dataset_audit(tmp_path: Path) -> None:
             str(tmp_path / "base.pt"),
             "--train-yaml",
             str(tmp_path / "dataset.yaml"),
+            "--train-summary",
+            str(tmp_path / "train-summary.json"),
             "--dataset-audit",
             str(audit),
             "--calibration-summary",