This commit is contained in:
Mateo Wang 2026-11-06 15:50:55 -05:00 • committed by GitHub
commit b744559201
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -282,11 +282,9 @@ def _expect_metrics(
blank: float,
*,
api_provider: str = "openai",
pf_id: str | None = None,
pf_populated: float | None = None,
pf_blank: float | None = None,
) -> tuple:
expected_id: Final = rig.deployment_id if pf_id is None else pf_id
expected_pf_populated: Final = populated if pf_populated is None else pf_populated
expected_pf_blank: Final = blank if pf_blank is None else pf_blank
@ -295,7 +293,7 @@ def _expect_metrics(
satisfied: Final = (
_populated_failures(samples, rig, api_provider) == populated
and _count(samples, DEPLOYMENT_FAILURE, "") == blank
and _count(samples, PROXY_FAILED, expected_id) == expected_pf_populated
and _count(samples, PROXY_FAILED, rig.deployment_id) == expected_pf_populated
and _count(samples, PROXY_FAILED, "") == expected_pf_blank
)
return samples if satisfied else ()