driftless 0.2.5__tar.gz → 0.2.6__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {driftless-0.2.5 → driftless-0.2.6}/CHANGELOG.md +13 -2
- {driftless-0.2.5 → driftless-0.2.6}/PKG-INFO +5 -5
- {driftless-0.2.5 → driftless-0.2.6}/README.md +4 -4
- {driftless-0.2.5 → driftless-0.2.6}/docs/RELEASE.md +7 -5
- {driftless-0.2.5 → driftless-0.2.6}/site/docs.html +1 -1
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/__init__.py +1 -1
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/contract.py +10 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/engine.py +42 -8
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/evaluation.py +33 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/report.py +1 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_contract.py +11 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_engine.py +10 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_evaluation.py +11 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_init_ci.py +40 -0
- driftless-0.2.6/tests/test_splits.py +27 -0
- {driftless-0.2.5 → driftless-0.2.6}/.gitignore +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/LICENSE +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/docs/repair-and-generators.md +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/pyproject.toml +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/site/assets/app.js +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/site/assets/hero-workflow.png +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/site/assets/landing.css +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/site/assets/runs.css +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/site/assets/runs.js +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/site/assets/sample-run.json +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/site/assets/styles.css +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/site/index.html +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/site/runs.html +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/calibrate.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/cli.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/compare.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/configure.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/data/model_lifecycle.json +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/datasource.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/datastate.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/discovery.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/errors.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/generators.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/github.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/harness.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/init_ci.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/judges.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/label_audit.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/lifecycle.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/policy.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/preflight.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/progress.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/scanner.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/splits.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/templates.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/src/driftless/view.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/fixtures/live_eval_baseline.json +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/fixtures/smoke/driftless.yml +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/fixtures/smoke/inputs.jsonl +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/fixtures/smoke/labels.jsonl +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/regression_metrics.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/scenarios.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_cli.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_compare.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_data_change_gate.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_data_change_regression.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_datasource.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_datastate.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_discovery.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_endpoint.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_extraction.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_generators.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_github.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_grading_loop.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_harness.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_judge.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_judge_loop.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_label_audit.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_lifecycle.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_migration_live.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_migration_regression.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_plan_act.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_policy.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_poll_act.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_preflight.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_progress.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_refine.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_refresh_catalog.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_regression_metrics.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_repair_prompt.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_report.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_scanner.py +0 -0
- {driftless-0.2.5 → driftless-0.2.6}/tests/test_view.py +0 -0
|
@@ -17,6 +17,16 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
|
|
|
17
17
|
|
|
18
18
|
---
|
|
19
19
|
|
|
20
|
+
## [0.2.6] - 2026-07-01
|
|
21
|
+
|
|
22
|
+
### Added
|
|
23
|
+
|
|
24
|
+
- **P0.3 multi-seed tuning selection** — optional `migration.split_seed_count`
|
|
25
|
+
(1–5) averages tuning-split metrics across shuffle seeds when scoring repair
|
|
26
|
+
candidates; holdout validation still uses the primary `--seed` only.
|
|
27
|
+
|
|
28
|
+
---
|
|
29
|
+
|
|
20
30
|
## [0.2.5] - 2026-07-01
|
|
21
31
|
|
|
22
32
|
### Added
|
|
@@ -132,8 +142,9 @@ First public release on [PyPI](https://pypi.org/project/driftless/0.1.0/).
|
|
|
132
142
|
- **Docs** — project overview, repair algorithm spec, 2×2 migration methodology,
|
|
133
143
|
Poetry + Dependabot product framing.
|
|
134
144
|
|
|
135
|
-
[Unreleased]: https://github.com/driftless-dev/driftless/compare/v0.2.
|
|
136
|
-
[0.2.
|
|
145
|
+
[Unreleased]: https://github.com/driftless-dev/driftless/compare/v0.2.6...HEAD
|
|
146
|
+
[0.2.6]: https://github.com/driftless-dev/driftless/releases/tag/v0.2.6
|
|
147
|
+
[0.2.5]: https://github.com/driftless-dev/driftless/compare/v0.2.5...v0.2.6
|
|
137
148
|
[0.2.4]: https://github.com/driftless-dev/driftless/compare/v0.2.4...v0.2.5
|
|
138
149
|
[0.2.3]: https://github.com/driftless-dev/driftless/compare/v0.2.3...v0.2.4
|
|
139
150
|
[0.2.2]: https://github.com/driftless-dev/driftless/compare/v0.2.2...v0.2.3
|
|
@@ -1,6 +1,6 @@
|
|
|
1
1
|
Metadata-Version: 2.4
|
|
2
2
|
Name: driftless
|
|
3
|
-
Version: 0.2.
|
|
3
|
+
Version: 0.2.6
|
|
4
4
|
Summary: Keep prompts in sync when model or eval data changes — Poetry-style lock regeneration, Dependabot-style PRs.
|
|
5
5
|
Project-URL: Homepage, https://github.com/driftless-dev/driftless
|
|
6
6
|
Project-URL: Repository, https://github.com/driftless-dev/driftless
|
|
@@ -96,7 +96,7 @@ optimizes against it, with your team owning the definition of "good":
|
|
|
96
96
|
|---|---|
|
|
97
97
|
| `init` | Scaffold a `driftless.yml`. |
|
|
98
98
|
| `init-policy` | Scaffold a `.driftless/policy.yml` (when to migrate). |
|
|
99
|
-
| `init-ci` | Scaffold `.github/workflows/` for scan, migrate, refine, poll, label audit, and judge check. |
|
|
99
|
+
| `init-ci` | Scaffold `.github/workflows/` for scan, migrate, refine, poll, plan, label audit, and judge check. |
|
|
100
100
|
| `scan` | Find probable LLM usage and at-risk models. |
|
|
101
101
|
| `plan` | Discover at-risk workflows and apply the migration policy (CI triage). |
|
|
102
102
|
| `plan --act` | Migrate + open a PR/issue for every actionable trigger (close the loop). |
|
|
@@ -129,11 +129,11 @@ propose it.
|
|
|
129
129
|
## GitHub-native usage
|
|
130
130
|
|
|
131
131
|
A composite GitHub Action (`action.yml`) wraps the CLI so scans and migrations
|
|
132
|
-
can run in CI. See `.github/workflows/` for a scheduled deprecation scan
|
|
133
|
-
manually-triggered migration
|
|
132
|
+
can run in CI. See `.github/workflows/` for a scheduled deprecation scan, weekly
|
|
133
|
+
`plan --act` triage, and manually-triggered migration workflows.
|
|
134
134
|
|
|
135
135
|
```yaml
|
|
136
|
-
- uses: driftless-dev/driftless@v0.2.
|
|
136
|
+
- uses: driftless-dev/driftless@v0.2.6
|
|
137
137
|
with:
|
|
138
138
|
command: scan
|
|
139
139
|
```
|
|
@@ -57,7 +57,7 @@ optimizes against it, with your team owning the definition of "good":
|
|
|
57
57
|
|---|---|
|
|
58
58
|
| `init` | Scaffold a `driftless.yml`. |
|
|
59
59
|
| `init-policy` | Scaffold a `.driftless/policy.yml` (when to migrate). |
|
|
60
|
-
| `init-ci` | Scaffold `.github/workflows/` for scan, migrate, refine, poll, label audit, and judge check. |
|
|
60
|
+
| `init-ci` | Scaffold `.github/workflows/` for scan, migrate, refine, poll, plan, label audit, and judge check. |
|
|
61
61
|
| `scan` | Find probable LLM usage and at-risk models. |
|
|
62
62
|
| `plan` | Discover at-risk workflows and apply the migration policy (CI triage). |
|
|
63
63
|
| `plan --act` | Migrate + open a PR/issue for every actionable trigger (close the loop). |
|
|
@@ -90,11 +90,11 @@ propose it.
|
|
|
90
90
|
## GitHub-native usage
|
|
91
91
|
|
|
92
92
|
A composite GitHub Action (`action.yml`) wraps the CLI so scans and migrations
|
|
93
|
-
can run in CI. See `.github/workflows/` for a scheduled deprecation scan
|
|
94
|
-
manually-triggered migration
|
|
93
|
+
can run in CI. See `.github/workflows/` for a scheduled deprecation scan, weekly
|
|
94
|
+
`plan --act` triage, and manually-triggered migration workflows.
|
|
95
95
|
|
|
96
96
|
```yaml
|
|
97
|
-
- uses: driftless-dev/driftless@v0.2.
|
|
97
|
+
- uses: driftless-dev/driftless@v0.2.6
|
|
98
98
|
with:
|
|
99
99
|
command: scan
|
|
100
100
|
```
|
|
@@ -153,7 +153,7 @@ After a release, users can pin the composite Action by release tag
|
|
|
153
153
|
(`action.yml` lives at the repo root — no `/action` path segment):
|
|
154
154
|
|
|
155
155
|
```yaml
|
|
156
|
-
- uses: driftless-dev/driftless@v0.2.
|
|
156
|
+
- uses: driftless-dev/driftless@v0.2.6
|
|
157
157
|
with:
|
|
158
158
|
command: scan
|
|
159
159
|
```
|
|
@@ -161,9 +161,9 @@ After a release, users can pin the composite Action by release tag
|
|
|
161
161
|
Or pin the PyPI package in the Action input:
|
|
162
162
|
|
|
163
163
|
```yaml
|
|
164
|
-
- uses: driftless-dev/driftless@v0.2.
|
|
164
|
+
- uses: driftless-dev/driftless@v0.2.6
|
|
165
165
|
with:
|
|
166
|
-
version: "==0.2.
|
|
166
|
+
version: "==0.2.6"
|
|
167
167
|
command: migrate
|
|
168
168
|
```
|
|
169
169
|
|
|
@@ -171,7 +171,7 @@ Optionally maintain a floating **`v1`** tag on the latest stable minor release
|
|
|
171
171
|
(point it at the current release tag after each publish):
|
|
172
172
|
|
|
173
173
|
```bash
|
|
174
|
-
git tag -f v1 v0.2.
|
|
174
|
+
git tag -f v1 v0.2.6 && git push origin v1 --force
|
|
175
175
|
```
|
|
176
176
|
|
|
177
177
|
Update [`action.yml`](../action.yml) default `version` input when cutting releases.
|
|
@@ -213,7 +213,9 @@ In **Settings → Secrets and variables → Actions**, add:
|
|
|
213
213
|
| `ANTHROPIC_API_KEY` | Live eval matrix job (`provider: anthropic`) |
|
|
214
214
|
|
|
215
215
|
If a secret is missing, that provider job exits cleanly with a warning (CI stays
|
|
216
|
-
green).
|
|
216
|
+
green). On scheduled or manual runs, the **secrets-preflight** job writes a
|
|
217
|
+
summary table to the workflow run so you can see which keys are configured.
|
|
218
|
+
When both are set, nightly runs append to
|
|
217
219
|
`.driftless/regression-metrics.jsonl` and check against
|
|
218
220
|
`tests/fixtures/live_eval_baseline.json` with `--require-all`.
|
|
219
221
|
|
|
@@ -428,7 +428,7 @@ driftless view -w support_classifier</code></pre>
|
|
|
428
428
|
<span class="tok-k">runs-on</span>: ubuntu-latest
|
|
429
429
|
<span class="tok-k">steps</span>:
|
|
430
430
|
- <span class="tok-k">uses</span>: actions/checkout@v4
|
|
431
|
-
- <span class="tok-k">uses</span>: driftless-dev/driftless@v0.2.
|
|
431
|
+
- <span class="tok-k">uses</span>: driftless-dev/driftless@v0.2.6
|
|
432
432
|
<span class="tok-k">with</span>:
|
|
433
433
|
<span class="tok-k">command</span>: <span class="tok-s">plan</span></code></pre>
|
|
434
434
|
<p>A scheduled <code class="inline">plan</code> gates CI when a deprecated model needs attention; a manually-triggered <code class="inline">migrate</code> opens a PR (or an issue when blocked) with the evidence attached.</p>
|
|
@@ -324,6 +324,16 @@ class MigrationSpec(StrictModel):
|
|
|
324
324
|
allow_business_logic_edits: bool = False
|
|
325
325
|
max_iterations: int = 8
|
|
326
326
|
holdout_required: bool = True
|
|
327
|
+
# When >1, average tuning-split metrics across this many shuffle seeds
|
|
328
|
+
# (seed, seed+1, …) when scoring repair candidates. Holdout uses ``seed`` only.
|
|
329
|
+
split_seed_count: int = 1
|
|
330
|
+
|
|
331
|
+
@field_validator("split_seed_count")
|
|
332
|
+
@classmethod
|
|
333
|
+
def _split_seed_count_range(cls, v: int) -> int:
|
|
334
|
+
if v < 1 or v > 5:
|
|
335
|
+
raise ValueError("migration.split_seed_count must be between 1 and 5")
|
|
336
|
+
return v
|
|
327
337
|
|
|
328
338
|
|
|
329
339
|
class RepairSpec(StrictModel):
|
|
@@ -30,10 +30,10 @@ from .calibrate import suggest_thresholds
|
|
|
30
30
|
from .compare import ThresholdCheck, check_thresholds
|
|
31
31
|
from .contract import ThresholdsSpec, Workflow
|
|
32
32
|
from .errors import DriftlessError
|
|
33
|
-
from .evaluation import Metrics, RecordRow, RunAnalysis, analyze
|
|
33
|
+
from .evaluation import Metrics, RecordRow, RunAnalysis, analyze, average_metrics
|
|
34
34
|
from .harness import run_workflow
|
|
35
35
|
from .progress import log as progress_log
|
|
36
|
-
from .splits import make_splits, materialize_inputs
|
|
36
|
+
from .splits import Split, make_splits, materialize_inputs
|
|
37
37
|
|
|
38
38
|
|
|
39
39
|
# --------------------------------------------------------------------------- #
|
|
@@ -336,6 +336,8 @@ class MigrationResult:
|
|
|
336
336
|
message: str = ""
|
|
337
337
|
# Frozen editable files at loop start — baseline for per-candidate diffs in reports/UI.
|
|
338
338
|
original_editable_files: dict[str, str] = field(default_factory=dict)
|
|
339
|
+
# Shuffle seeds used for tuning (primary ``seed`` only when split_seed_count==1).
|
|
340
|
+
split_seeds_used: list[int] = field(default_factory=list)
|
|
339
341
|
|
|
340
342
|
@property
|
|
341
343
|
def succeeded(self) -> bool:
|
|
@@ -516,11 +518,19 @@ def run_migration(
|
|
|
516
518
|
)
|
|
517
519
|
|
|
518
520
|
split = make_splits(workflow, cwd=cwd, seed=seed)
|
|
521
|
+
split_seeds_used = list(range(seed, seed + mig.split_seed_count))
|
|
519
522
|
size_warnings = assess_split_sizes(
|
|
520
523
|
len(split.input_lines),
|
|
521
524
|
len(split.holdout_idx),
|
|
522
525
|
holdout_required=mig.holdout_required,
|
|
523
526
|
)
|
|
527
|
+
if mig.split_seed_count > 1:
|
|
528
|
+
size_warnings.append(
|
|
529
|
+
f"Multi-seed tuning: candidate selection averages metrics across "
|
|
530
|
+
f"{mig.split_seed_count} shuffle seeds ({split_seeds_used[0]}.."
|
|
531
|
+
f"{split_seeds_used[-1]}); each candidate scoring multiplies tuning "
|
|
532
|
+
"workflow runs."
|
|
533
|
+
)
|
|
524
534
|
|
|
525
535
|
use_ids = bool(workflow.eval.id_field) and split.gold is not None
|
|
526
536
|
|
|
@@ -532,18 +542,23 @@ def run_migration(
|
|
|
532
542
|
return judge_evidence_samples(rows)
|
|
533
543
|
|
|
534
544
|
def evaluate_on(
|
|
535
|
-
model: str,
|
|
545
|
+
model: str,
|
|
546
|
+
idx: list[int],
|
|
547
|
+
files: dict[str, str] | None = None,
|
|
548
|
+
*,
|
|
549
|
+
split_ref: Split | None = None,
|
|
536
550
|
) -> RunAnalysis:
|
|
551
|
+
sp = split_ref or split
|
|
537
552
|
file_ctx = apply_files(files, cwd=cwd) if files else nullcontext()
|
|
538
|
-
idx_lines =
|
|
553
|
+
idx_lines = sp.lines_for(idx)
|
|
539
554
|
with materialize_inputs(workflow, idx_lines, cwd=cwd):
|
|
540
555
|
with file_ctx:
|
|
541
556
|
run = run_workflow(workflow, model, cwd=cwd)
|
|
542
|
-
if use_ids:
|
|
557
|
+
if use_ids and sp.gold_ids is not None:
|
|
543
558
|
return analyze(
|
|
544
559
|
workflow,
|
|
545
560
|
run,
|
|
546
|
-
gold_by_id=
|
|
561
|
+
gold_by_id=sp.gold_by_id_for(idx),
|
|
547
562
|
inputs=idx_lines,
|
|
548
563
|
judge=judge,
|
|
549
564
|
cwd=cwd,
|
|
@@ -551,12 +566,27 @@ def run_migration(
|
|
|
551
566
|
return analyze(
|
|
552
567
|
workflow,
|
|
553
568
|
run,
|
|
554
|
-
gold_labels=
|
|
569
|
+
gold_labels=sp.gold_for(idx),
|
|
555
570
|
inputs=idx_lines,
|
|
556
571
|
judge=judge,
|
|
557
572
|
cwd=cwd,
|
|
558
573
|
)
|
|
559
574
|
|
|
575
|
+
def evaluate_tuning(
|
|
576
|
+
model: str, files: dict[str, str] | None = None
|
|
577
|
+
) -> RunAnalysis:
|
|
578
|
+
"""Score on the tuning split; average across seeds when configured."""
|
|
579
|
+
if mig.split_seed_count <= 1:
|
|
580
|
+
return evaluate_on(model, split.tuning_idx, files)
|
|
581
|
+
tuning_splits = [make_splits(workflow, cwd=cwd, seed=s) for s in split_seeds_used]
|
|
582
|
+
analyses = [
|
|
583
|
+
evaluate_on(model, sp.tuning_idx, files, split_ref=sp) for sp in tuning_splits
|
|
584
|
+
]
|
|
585
|
+
return RunAnalysis(
|
|
586
|
+
metrics=average_metrics([a.metrics for a in analyses]),
|
|
587
|
+
rows=analyses[0].rows,
|
|
588
|
+
)
|
|
589
|
+
|
|
560
590
|
progress_log(
|
|
561
591
|
f"migration: phase 1/3 — initial eval "
|
|
562
592
|
f"({len(split.tuning_idx)} tuning examples, model={current})"
|
|
@@ -597,6 +627,7 @@ def run_migration(
|
|
|
597
627
|
holdout_checks=holdout_checks,
|
|
598
628
|
tuning_checks=naive_checks,
|
|
599
629
|
warnings=size_warnings,
|
|
630
|
+
split_seeds_used=split_seeds_used,
|
|
600
631
|
judge_agreement=judge_agreement_info,
|
|
601
632
|
judge_evidence=_judge_evidence(naive_analysis.rows),
|
|
602
633
|
message="naive model swap passes thresholds; only the model ID changes",
|
|
@@ -691,7 +722,7 @@ def run_migration(
|
|
|
691
722
|
cand_size = _patch_diff_size(patch.files, original_editable)
|
|
692
723
|
try:
|
|
693
724
|
validate_patch_scope(patch, workflow, cwd)
|
|
694
|
-
analysis =
|
|
725
|
+
analysis = evaluate_tuning(target_model, files=patch.files)
|
|
695
726
|
except DriftlessError as exc:
|
|
696
727
|
experiment_log.append(
|
|
697
728
|
AttemptRecord(
|
|
@@ -786,6 +817,7 @@ def run_migration(
|
|
|
786
817
|
experiment_log=experiment_log,
|
|
787
818
|
cluster_history=cluster_history,
|
|
788
819
|
warnings=size_warnings,
|
|
820
|
+
split_seeds_used=split_seeds_used,
|
|
789
821
|
judge_agreement=judge_agreement_info,
|
|
790
822
|
judge_evidence=_judge_evidence(best_analysis.rows),
|
|
791
823
|
original_editable_files=original_editable,
|
|
@@ -855,6 +887,7 @@ def run_migration(
|
|
|
855
887
|
experiment_log=experiment_log,
|
|
856
888
|
cluster_history=cluster_history,
|
|
857
889
|
warnings=size_warnings,
|
|
890
|
+
split_seeds_used=split_seeds_used,
|
|
858
891
|
suggested_thresholds=suggested,
|
|
859
892
|
judge_agreement=judge_agreement_info,
|
|
860
893
|
judge_evidence=_judge_evidence(best_analysis.rows),
|
|
@@ -887,6 +920,7 @@ def run_migration(
|
|
|
887
920
|
experiment_log=experiment_log,
|
|
888
921
|
cluster_history=cluster_history,
|
|
889
922
|
warnings=size_warnings,
|
|
923
|
+
split_seeds_used=split_seeds_used,
|
|
890
924
|
judge_agreement=judge_agreement_info,
|
|
891
925
|
judge_evidence=_judge_evidence(best_analysis.rows),
|
|
892
926
|
original_editable_files=original_editable,
|
|
@@ -96,6 +96,39 @@ class Metrics:
|
|
|
96
96
|
scored: int = 0
|
|
97
97
|
|
|
98
98
|
|
|
99
|
+
def average_metrics(items: list[Metrics]) -> Metrics:
|
|
100
|
+
"""Mean of headline metrics across multiple eval runs (multi-seed tuning)."""
|
|
101
|
+
if not items:
|
|
102
|
+
raise ValueError("average_metrics requires at least one Metrics")
|
|
103
|
+
if len(items) == 1:
|
|
104
|
+
return items[0]
|
|
105
|
+
|
|
106
|
+
def _mean(vals: list[float | None]) -> float | None:
|
|
107
|
+
nums = [v for v in vals if v is not None]
|
|
108
|
+
return sum(nums) / len(nums) if nums else None
|
|
109
|
+
|
|
110
|
+
def _mean_int(vals: list[int]) -> int:
|
|
111
|
+
return int(round(sum(vals) / len(vals)))
|
|
112
|
+
|
|
113
|
+
costs = [m.total_cost for m in items if m.total_cost is not None]
|
|
114
|
+
return Metrics(
|
|
115
|
+
n=items[0].n,
|
|
116
|
+
schema_error_rate=_mean([m.schema_error_rate for m in items]),
|
|
117
|
+
refusal_rate=_mean([m.refusal_rate for m in items]) or 0.0,
|
|
118
|
+
accuracy=_mean([m.accuracy for m in items]),
|
|
119
|
+
precision=_mean([m.precision for m in items]),
|
|
120
|
+
recall=_mean([m.recall for m in items]),
|
|
121
|
+
f1=_mean([m.f1 for m in items]),
|
|
122
|
+
avg_latency_ms=_mean([m.avg_latency_ms for m in items]),
|
|
123
|
+
total_cost=sum(costs) if costs else None,
|
|
124
|
+
score=_mean([m.score for m in items]),
|
|
125
|
+
schema_errors=_mean_int([m.schema_errors for m in items]),
|
|
126
|
+
refusals=_mean_int([m.refusals for m in items]),
|
|
127
|
+
labeled=items[0].labeled,
|
|
128
|
+
scored=items[0].scored,
|
|
129
|
+
)
|
|
130
|
+
|
|
131
|
+
|
|
99
132
|
def load_jsonl(path: Path) -> list[OutputRecord]:
|
|
100
133
|
records: list[OutputRecord] = []
|
|
101
134
|
with path.open(encoding="utf-8") as fh:
|
|
@@ -568,6 +568,7 @@ def result_to_dict(result: MigrationResult) -> dict:
|
|
|
568
568
|
"experiment_log": [asdict(a) for a in result.experiment_log],
|
|
569
569
|
"cluster_trajectory": cluster_trajectories(result.cluster_history),
|
|
570
570
|
"warnings": result.warnings,
|
|
571
|
+
"split_seeds_used": result.split_seeds_used,
|
|
571
572
|
"judge_agreement": asdict(result.judge_agreement) if result.judge_agreement else None,
|
|
572
573
|
"judge_evidence": result.judge_evidence,
|
|
573
574
|
"suggested_thresholds": result.suggested_thresholds,
|
|
@@ -63,6 +63,17 @@ def test_workflow_not_found():
|
|
|
63
63
|
contract.workflow("missing")
|
|
64
64
|
|
|
65
65
|
|
|
66
|
+
def test_split_seed_count_must_be_in_range():
|
|
67
|
+
with pytest.raises(Exception):
|
|
68
|
+
Workflow.model_validate(
|
|
69
|
+
{
|
|
70
|
+
"run": {"command": "true", "input_path": "i", "output_path": "o"},
|
|
71
|
+
"model": {"current": "m", "env_var": "M"},
|
|
72
|
+
"migration": {"split_seed_count": 0},
|
|
73
|
+
}
|
|
74
|
+
)
|
|
75
|
+
|
|
76
|
+
|
|
66
77
|
def test_load_missing_contract(tmp_path: Path):
|
|
67
78
|
with pytest.raises(ContractError):
|
|
68
79
|
load_contract(tmp_path / "nope.yml")
|
|
@@ -207,3 +207,13 @@ def test_cluster_failures():
|
|
|
207
207
|
assert kinds["refusal"].count == 1
|
|
208
208
|
assert kinds["misclassification"].count == 2 # billing<-technical pair
|
|
209
209
|
assert kinds["misclassification"].key == "billing -> technical"
|
|
210
|
+
|
|
211
|
+
|
|
212
|
+
def test_multi_seed_tuning_still_passes(tmp_path: Path):
|
|
213
|
+
wf = _make_workflow(tmp_path)
|
|
214
|
+
wf.migration.split_seed_count = 2
|
|
215
|
+
result = run_migration("demo", wf, "weak", generator=StrictGen(), cwd=tmp_path, seed=1)
|
|
216
|
+
|
|
217
|
+
assert result.status == MigrationStatus.PASS
|
|
218
|
+
assert result.split_seeds_used == [1, 2]
|
|
219
|
+
assert any("Multi-seed tuning" in w for w in result.warnings)
|
|
@@ -316,3 +316,14 @@ def test_load_labels_by_id_rejects_duplicates(tmp_path: Path):
|
|
|
316
316
|
p.write_text('{"id":"a","label":"x"}\n{"id":"a","label":"y"}\n')
|
|
317
317
|
with pytest.raises(Exception):
|
|
318
318
|
load_labels_by_id(p, "id", "label")
|
|
319
|
+
|
|
320
|
+
|
|
321
|
+
def test_average_metrics_means_headline_fields():
|
|
322
|
+
from driftless.evaluation import Metrics, average_metrics
|
|
323
|
+
|
|
324
|
+
a = Metrics(n=10, schema_error_rate=0.2, refusal_rate=0.1, f1=0.8)
|
|
325
|
+
b = Metrics(n=10, schema_error_rate=0.0, refusal_rate=0.0, f1=1.0)
|
|
326
|
+
avg = average_metrics([a, b])
|
|
327
|
+
assert avg.f1 == pytest.approx(0.9)
|
|
328
|
+
assert avg.schema_error_rate == pytest.approx(0.1)
|
|
329
|
+
assert avg.refusal_rate == pytest.approx(0.05)
|
|
@@ -293,6 +293,46 @@ def test_label_audit_helpers():
|
|
|
293
293
|
assert label_audit_paths(contract) == ["labels.jsonl", "in.jsonl"]
|
|
294
294
|
|
|
295
295
|
|
|
296
|
+
def test_init_ci_scaffolds_plan_workflow(tmp_path, monkeypatch):
|
|
297
|
+
monkeypatch.chdir(tmp_path)
|
|
298
|
+
Path("driftless.yml").write_text(
|
|
299
|
+
"""
|
|
300
|
+
version: 1
|
|
301
|
+
workflows:
|
|
302
|
+
smoke:
|
|
303
|
+
run:
|
|
304
|
+
command: echo ok
|
|
305
|
+
input_path: in.jsonl
|
|
306
|
+
output_path: out.jsonl
|
|
307
|
+
model:
|
|
308
|
+
current: gpt-4o-mini
|
|
309
|
+
env_var: MODEL
|
|
310
|
+
eval:
|
|
311
|
+
labels_path: labels.jsonl
|
|
312
|
+
""".lstrip()
|
|
313
|
+
)
|
|
314
|
+
out = tmp_path / "workflows"
|
|
315
|
+
result = runner.invoke(
|
|
316
|
+
app,
|
|
317
|
+
[
|
|
318
|
+
"init-ci",
|
|
319
|
+
"--out-dir",
|
|
320
|
+
str(out),
|
|
321
|
+
"--no-scan",
|
|
322
|
+
"--no-migrate",
|
|
323
|
+
"--no-refine",
|
|
324
|
+
"--no-audit-labels",
|
|
325
|
+
"--plan",
|
|
326
|
+
],
|
|
327
|
+
)
|
|
328
|
+
|
|
329
|
+
assert result.exit_code == 0
|
|
330
|
+
plan = (out / "driftless-plan-act.yml").read_text()
|
|
331
|
+
assert "command: plan" in plan
|
|
332
|
+
assert "--act" in plan
|
|
333
|
+
assert "GH_TOKEN" in plan
|
|
334
|
+
|
|
335
|
+
|
|
296
336
|
def test_rendered_workflows_use_action_ref():
|
|
297
337
|
ref = "driftless-dev/driftless@v9.9.9"
|
|
298
338
|
assert ref in render_migrate_workflow(ref)
|
|
@@ -0,0 +1,27 @@
|
|
|
1
|
+
"""Tests for tuning/holdout splits."""
|
|
2
|
+
|
|
3
|
+
from driftless.contract import Workflow
|
|
4
|
+
from driftless.splits import make_splits
|
|
5
|
+
|
|
6
|
+
|
|
7
|
+
def _workflow() -> Workflow:
|
|
8
|
+
return Workflow.model_validate(
|
|
9
|
+
{
|
|
10
|
+
"run": {"command": "true", "input_path": "i.jsonl", "output_path": "o.jsonl"},
|
|
11
|
+
"model": {"current": "m", "env_var": "M"},
|
|
12
|
+
"eval": {"labels_path": "l.jsonl", "split": {"tuning": 0.5, "holdout": 0.5}},
|
|
13
|
+
}
|
|
14
|
+
)
|
|
15
|
+
|
|
16
|
+
|
|
17
|
+
def test_different_seeds_produce_different_partitions(tmp_path):
|
|
18
|
+
lines = "\n".join(f'{{"id": {i}, "label": "a"}}' for i in range(20)) + "\n"
|
|
19
|
+
labels = "\n".join('{"id": ' + str(i) + ', "label": "a"}' for i in range(20)) + "\n"
|
|
20
|
+
(tmp_path / "i.jsonl").write_text(lines)
|
|
21
|
+
(tmp_path / "l.jsonl").write_text(labels)
|
|
22
|
+
|
|
23
|
+
wf = _workflow()
|
|
24
|
+
wf.eval.id_field = "id"
|
|
25
|
+
a = make_splits(wf, cwd=tmp_path, seed=0)
|
|
26
|
+
b = make_splits(wf, cwd=tmp_path, seed=1)
|
|
27
|
+
assert a.tuning_idx != b.tuning_idx
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|